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

fix

main
ifiguero 2025-03-18 15:01:35 -03:00
parent 24375b392b
commit 757946c078
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
TrainerClass.py
View File

@ -488,6 +488,7 @@ class eNoseTrainer:
df.to_excel(f"{model_file}.search.xlsx", index=False)
self.logger.info(f"Training Model {model_id} with {study.best_params}")
early_stopping = keras.callbacks.EarlyStopping(monitor='loss', patience=5, restore_best_weights=True, min_delta=0.0003)
best_model.fit(X_train, Y_train, epochs=epochs, batch_size=best_batch_size, verbose=1, callbacks=[early_stopping])
@ -578,7 +579,8 @@ class eNoseTrainer:
df.to_excel(f"{model_file}.search.xlsx", index=False)
self.logger.info(f"Training Model {model_id} with {study.best_params}")
early_stopping = keras.callbacks.EarlyStopping(monitor='loss', patience=5, restore_best_weights=True)
early_stopping = keras.callbacks.EarlyStopping(monitor='loss', patience=5, restore_best_weights=True, min_delta=0.0003)
best_model.fit(X_train, Y_train, epochs=epochs, batch_size=best_batch_size, verbose=1, callbacks=[early_stopping])
best_model.save(f"{model_file}.keras")