From 3799786c37623e1689d985dcb17d3d367ce3f4f0 Mon Sep 17 00:00:00 2001 From: Israel Figueroa Date: Mon, 10 Mar 2025 18:52:21 -0300 Subject: [PATCH] patch --- TrainerClass.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/TrainerClass.py b/TrainerClass.py index 7422dca..9be995c 100644 --- 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()