Merge pull request #67 from wannaphong/fix-train

Fixed train.py
rkcosmos · Oct 23, 2020 · deb4008 · deb4008
2 parents 37712bb + 18e6020
commit deb4008
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/deepcut/train.py b/deepcut/train.py
@@ -135,9 +135,9 @@ def prepare_feature(best_processed_path, option='train'):
     type_row = ['type' + str(i + 1) for i in range(n_pad_2)] + \
                ['type-' + str(i + 1) for i in range(n_pad_2)] + ['type']
 
-    x_char = df_pad[char_row].as_matrix()
-    x_type = df_pad[type_row].as_matrix()
-    y = df_pad['target'].astype(int).as_matrix()
+    x_char = df_pad[char_row].to_numpy()
+    x_type = df_pad[type_row].to_numpy()
+    y = df_pad['target'].astype(int).to_numpy()
 
     return x_char, x_type, y