Jérôme BUISINE il y a 3 ans
Parent
commit
1e89cf4438
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  1. 1 1
      train_model.py

+ 1 - 1
train_model.py

@@ -270,7 +270,7 @@ def main():
         callbacks=callbacks_list, 
         class_weight=class_weight)
 
-    score = model.evaluate(X_val, y_val, batch_size=p_batch_size)
+    score = model.evaluate(X_val, y_val_cat, batch_size=p_batch_size)
 
     print("Accuracy score on val dataset ", score)