Commit 8e546668 authored by lucas_miranda's avatar lucas_miranda
Browse files

Updated default GMVAE models

parent 3243a42f
Pipeline #87795 passed with stage
in 17 minutes and 42 seconds
......@@ -422,7 +422,13 @@ else:
pheno_class=pheno_class,
predictor=predictor,
project_name="{}-based_{}_BAYESIAN_OPT".format(input_type, hyp),
callbacks=[tensorboard_callback, onecycle],
callbacks=[
tensorboard_callback,
onecycle,
tf.keras.callbacks.EarlyStopping(
"val_mae", patience=15, restore_best_weights=True
),
],
n_replicas=3,
n_epochs=30,
)
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment