patch

2025-03-10 18:52:21 -03:00 · 2025-03-10 18:52:21 -03:00 · 3799786c37
parent 4d2a32e75d
commit 3799786c37
1 changed files with 1 additions and 1 deletions
--- a/TrainerClass.py
+++ b/TrainerClass.py
@ -192,7 +192,7 @@ class eNoseTrainer:
        return tmse, mse, mae, rmse, optimized_model, model_params

    def fit(self):
-        total_train_queue = self.splits*len(self.get_model_train())
+        total_train_queue = int(1/self.ratio)*len(self.get_model_train())
        self.logger.info("{:=^60}".format(f'Begin Fit {total_train_queue} Models'))
        self.trained = 0
        manager = enlighten.get_manager()