Jérôme BUISINE 3 anni fa
parent
commit
1e89cf4438
1 ha cambiato i file con 1 aggiunte e 1 eliminazioni
  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)