From 6440021fe76b01feff3d5202f21cbc18dccad445 Mon Sep 17 00:00:00 2001 From: Israel Figueroa Date: Sun, 16 Mar 2025 23:22:18 -0300 Subject: [PATCH] test --- TrainerClass.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/TrainerClass.py b/TrainerClass.py index 77c393d..06204e0 100644 --- a/TrainerClass.py +++ b/TrainerClass.py @@ -213,15 +213,15 @@ class eNoseTrainer: output_dim = Y_trainc1D.shape[1] model = build_model_conv1D(config, input_shape, output_dim) - early_stopping = keras.callbacks.EarlyStopping(monitor='val_loss', patience=5, restore_best_weights=True) +# early_stopping = keras.callbacks.EarlyStopping(monitor='val_loss', patience=5, restore_best_weights=True) model.fit( X_trainc1D, Y_trainc1D, validation_data=(X_testc1D, Y_testc1D), epochs=config['epochs'], batch_size=config['batch_size'], - verbose=0, - callbacks=[early_stopping] + verbose=0 +# callbacks=[early_stopping] ) Y_pred = model.predict(X_testc1D)