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

test

main
ifiguero 2025-03-16 23:22:18 -03:00
parent edd035ef19
commit 6440021fe7
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
TrainerClass.py
View File

@ -213,15 +213,15 @@ class eNoseTrainer:
output_dim = Y_trainc1D.shape[1] output_dim = Y_trainc1D.shape[1]
model = build_model_conv1D(config, input_shape, output_dim) 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( model.fit(
X_trainc1D, Y_trainc1D, X_trainc1D, Y_trainc1D,
validation_data=(X_testc1D, Y_testc1D), validation_data=(X_testc1D, Y_testc1D),
epochs=config['epochs'], epochs=config['epochs'],
batch_size=config['batch_size'], batch_size=config['batch_size'],
verbose=0, verbose=0
callbacks=[early_stopping] # callbacks=[early_stopping]
) )
Y_pred = model.predict(X_testc1D) Y_pred = model.predict(X_testc1D)