Fixed IntuitionEngineeringTeam#8

dnishiyama · Apr 12, 2021 · ce09780 · ce09780
1 parent bccc3c4
commit ce09780
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/chars2vec/model.py b/chars2vec/model.py
@@ -233,7 +233,9 @@ def train_model(emb_dim, X_train, y_train, model_chars,
     char_to_ix = {ch: i for i, ch in enumerate(model_chars)}
     c2v_model = Chars2Vec(emb_dim, char_to_ix)
 
-    targets = [float(el) for el in y_train]
+    # TODO: https://github.com/IntuitionEngineeringTeam/chars2vec/issues/8
+    # targets = [float(el) for el in y_train]
+    targets = np.array(y_train)
     c2v_model.fit(X_train, targets, max_epochs, patience, validation_split, batch_size)
 
     return c2v_model