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

fix

main
ifiguero 2025-03-18 11:04:36 -03:00
parent 3557df0648
commit c7ba91be06
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
TrainerClass.py
View File

@ -578,7 +578,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='val_loss', patience=5, restore_best_weights=True)
early_stopping = keras.callbacks.EarlyStopping(monitor='loss', patience=5, restore_best_weights=True)
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")