diff --git a/TrainerClass.py b/TrainerClass.py index c867bd0..8087b62 100644 --- a/TrainerClass.py +++ b/TrainerClass.py @@ -252,7 +252,7 @@ class eNoseTrainer: ray.internal.free([X_train_ref, Y_train_ref, X_test_ref, Y_test_ref]) ray.shutdown() - return best_model, best_config + return best_model, analysis def train_and_score_model(self, model, X_train, X_test, Y_train, Y_test): param_dist = self.get_tunable_params(model) @@ -531,7 +531,9 @@ class eNoseTrainer: self.logger.debug(f"Y_train_sample: {Y_train_sample.shape}") self.logger.debug(f"Y_test_sample: {Y_test_sample.shape}") - optimized_model, model_params = self.search_best_conv1D_v1(X_train_sample, X_test_sample, Y_train_sample, Y_test_sample, epochs=epochs//3) + optimized_model, analysis = self.search_best_conv1D_v1(X_train_sample, X_test_sample, Y_train_sample, Y_test_sample, epochs=epochs//3) + model_params = analysis.get_best_config(metric='mse', mode='min') + analysis.results_df.to_excel(f"{model_file}.search.xlsx", index=False) self.logger.info(f"Training Model {model_id} with {model_params}") optimized_model.fit(X_train, Y_train, epochs=epochs, batch_size=model_params['batch_size'], verbose=1)