fixing right parameters to the report.

scaleoutsystems · Jun 25, 2024 · 9188b0f · 9188b0f
1 parent f1ad060
commit 9188b0f
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/Power-consumption-keras/client/validate.py b/Power-consumption-keras/client/validate.py
@@ -39,17 +39,13 @@ def validate(in_model_path, out_json_path, data_path=None):
     model.set_weights(weights)
 
     # Evaluate
-    model_score = model.evaluate(x_train, y_train)
+    model_score_train = model.evaluate(x_train, y_train)
     model_score_test = model.evaluate(x_test, y_test)
-    y_pred = model.predict(x_test)
-    y_pred = np.argmax(y_pred, axis=1)
 
     # JSON schema
     report = {
-        "training_loss": model_score[0],
-        "training_accuracy": model_score[1],
-        "test_loss": model_score_test[0],
-        "test_accuracy": model_score_test[1],
+        "training_mse": model_score_train[0],
+        "test_mse": model_score_test[0]
     }
 
     # Save JSON