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

fix

main
ifiguero 2025-03-18 22:49:04 -03:00
parent 70aa3e905d
commit 40f905c28d
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
TrainerClass.py
View File

@ -372,7 +372,7 @@ class eNoseTrainer:
"mse": mse,
"mae": mae,
"rmse": rmse,
"num_params": sum(t.count("\n") for t in optimized_model.get_booster().get_dump())
"num_params": sum(t.count("\n") for t in optimized_model.get_booster().get_dump(dump_format="json"))
}] )
self.ledger = pd.concat([self.ledger, newrow], ignore_index=True)
self.bar.update()
@ -430,7 +430,8 @@ class eNoseTrainer:
"mse": mse,
"mae": mae,
"rmse": rmse,
"num_params": sum(t.count("\n") for t in optimized_model.get_booster().get_dump())
"num_params": sum(t.count("\n") for t in optimized_model.get_booster().get_dump(dump_format="json"))
}] )
self.ledger = pd.concat([self.ledger, newrow], ignore_index=True)
self.bar.update()