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()