diff --git a/TrainerClass.py b/TrainerClass.py index f0ccf1d..a5b310d 100644 --- a/TrainerClass.py +++ b/TrainerClass.py @@ -525,7 +525,7 @@ class eNoseTrainer: optimized_model, model_params = self.search_best_conv1D_v1(X_train_sample, X_test_sample, Y_train_sample, Y_test_sample) - self.logger.info(f"Training Model {model_id} with {model_params}")) + self.logger.info(f"Training Model {model_id} with {model_params}") optimized_model.fit(X_train, Y_train, epochs=model_params['epochs'], batch_size=model_params['batch_size'], verbose=1) Y_train_pred = optimized_model.predict(X_train) @@ -601,6 +601,7 @@ class eNoseTrainer: optimized_model, model_params = self.search_best_conv1D_v1(X_train_sample, X_test_sample, Y_train_sample, Y_test_sample) + self.logger.info(f"Training Model {model_id} with {model_params}") optimized_model.fit(X_train, Y_train, epochs=model_params['epochs'], batch_size=model_params['batch_size'], verbose=1) Y_train_pred = optimized_model.predict(X_train)