From 40f905c28d08fbf75932ab51b2842ef616ef56bc Mon Sep 17 00:00:00 2001 From: Israel Figueroa Date: Tue, 18 Mar 2025 22:49:04 -0300 Subject: [PATCH] fix --- TrainerClass.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/TrainerClass.py b/TrainerClass.py index 4f795fd..237ac41 100644 --- a/TrainerClass.py +++ b/TrainerClass.py @@ -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()