DIA
/
enose_2025
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

saved

main
ifiguero 2025-03-16 23:18:03 -03:00
parent 013e49abe3
commit 5768fe2e1d
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
TrainerClass.py
View File

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