Commit e240953e authored by lucas_miranda's avatar lucas_miranda
Browse files

Changed default hyperparameter values

parent 054d3359
......@@ -428,7 +428,7 @@ else:
)
# Saves a compiled, untrained version of the best model
best_model.build(input_dict_train[input_type].shape)
best_model.build(X_train.shape)
best_model.save(
"{}-based_{}_BAYESIAN_OPT.h5".format(input_type, hyp), save_format="tf"
)
......
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