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

fix

main
ifiguero 2025-03-18 23:06:10 -03:00
parent 40f905c28d
commit d23ba450de
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
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(dump_format="json"))
"num_params": None if model_params.get('multi_strategy') == 'multi_output_tree' else sum(t.count("\n") for t in optimized_model.get_booster().get_dump())
}] )
self.ledger = pd.concat([self.ledger, newrow], ignore_index=True)
self.bar.update()
@ -430,7 +430,7 @@ class eNoseTrainer:
"mse": mse,
"mae": mae,
"rmse": rmse,
"num_params": sum(t.count("\n") for t in optimized_model.get_booster().get_dump(dump_format="json"))
"num_params": None if model_params.get('multi_strategy') == 'multi_output_tree' else sum(t.count("\n") for t in optimized_model.get_booster().get_dump())
}] )
self.ledger = pd.concat([self.ledger, newrow], ignore_index=True)