DIA
/
enose_2025
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

submit

main
ifiguero 2025-03-18 09:29:23 -03:00
parent dc8e27abe8
commit 8de4ee579c
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
TrainerClass.py
View File

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