Changed to python3

.idea/workspace.xml

@@ -34,8 +34,8 @@
       <file leaf-file-name="prepare_data.py" pinned="false" current-in-tab="true">
         <entry file="file://$PROJECT_DIR$/prepare_data.py">
           <provider selected="true" editor-type-id="text-editor">
-            <state relative-caret-position="606">
+            <state relative-caret-position="288">
-              <caret line="606" column="50" lean-forward="true" selection-start-line="606" selection-start-column="4" selection-end-line="606" selection-end-column="50" />
+              <caret line="16" column="0" lean-forward="true" selection-start-line="16" selection-start-column="0" selection-end-line="16" selection-end-column="0" />
               <folding>
                 <element signature="e#24#63#0" expanded="true" />
               </folding>
@@ -134,6 +134,7 @@
       <find>h5f</find>
       <find>generate_inputs</find>
       <find>split_number</find>
+      <find>StringIO</find>
     </findStrings>
   </component>
   <component name="Git.Settings">
@@ -153,10 +154,10 @@
     </option>
   </component>
   <component name="ProjectFrameBounds">
-    <option name="x" value="1985" />
+    <option name="x" value="65" />
     <option name="y" value="24" />
     <option name="width" value="1855" />
-    <option name="height" value="1176" />
+    <option name="height" value="1056" />
   </component>
   <component name="ProjectView">
     <navigator currentView="ProjectPane" proportions="" version="1">
@@ -173,8 +174,8 @@
       <foldersAlwaysOnTop value="true" />
     </navigator>
     <panes>
-      <pane id="Scope" />
       <pane id="Scratches" />
+      <pane id="Scope" />
       <pane id="ProjectPane">
         <subPane>
           <PATH>
@@ -482,7 +483,7 @@
     <servers />
   </component>
   <component name="ToolWindowManager">
-    <frame x="1985" y="24" width="1855" height="1176" extended-state="6" />
+    <frame x="65" y="24" width="1855" height="1056" extended-state="6" />
     <editor active="true" />
     <layout>
       <window_info id="Project" active="false" anchor="left" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="true" show_stripe_button="true" weight="0.16375546" sideWeight="0.5" order="0" side_tool="false" content_ui="combo" />
@@ -1009,8 +1010,8 @@
     </entry>
     <entry file="file://$PROJECT_DIR$/prepare_data.py">
       <provider selected="true" editor-type-id="text-editor">
-        <state relative-caret-position="606">
+        <state relative-caret-position="288">
-          <caret line="606" column="50" lean-forward="true" selection-start-line="606" selection-start-column="4" selection-end-line="606" selection-end-column="50" />
+          <caret line="16" column="0" lean-forward="true" selection-start-line="16" selection-start-column="0" selection-end-line="16" selection-end-column="0" />
           <folding>
             <element signature="e#24#63#0" expanded="true" />
           </folding>

prepare_data.py

@@ -6,7 +6,6 @@ import numpy as np
 import h5py
 import gc
 import math
-import StringIO
 import copy
 
 def save_inputs(file_name, X, y):
@@ -131,9 +130,9 @@ def create_dict():
                     dictionary.append(c)
             if num_vowels > max_num_vowels:
                 max_num_vowels = num_vowels
-        except Exception, e:
+        except Exception:
-            print line - 1
+            print(line - 1)
-            print el
+            print(el)
             break
         line += 1
     dictionary = sorted(dictionary)
@@ -394,7 +393,7 @@ def generate_X_and_y_RAM_efficient(name, split_number):
                     current_part_generation += 1
                 num_all_vowels += 1
         if i%10000 == 0:
-            print i
+            print(i)
         # text_file.write("Purchase Amount: %s" % TotalAmount)
         j = 0
         # X.append(word)