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

typo

main
ifiguero 2025-03-16 21:46:34 -03:00
parent 5e87e9b8f2
commit 714fa2e87e
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
TrainerClass.py
View File

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