submit

TrainerClass.py

@@ -601,11 +601,12 @@ class eNoseTrainer:
                                     "Ratio": self.ratio,
                                     "Model": model_id,
                                     "Params": json.dumps(study.best_params),
-                                    "Train mse": tmse,
-                                    "mse": mse,
-                                    "mae": mae,
-                                    "rmse": rmse
+                                    "Train mse": mse_train,
+                                    "mse": mse_test,
+                                    "mae": mae_test,
+                                    "rmse": rmse_test
                                     }] )
+                                    
                 self.ledger = pd.concat([self.ledger, newrow], ignore_index=True)
                 self.bar.update()
                 self.saveCheckPoint()