Skip to content
Snippets Groups Projects
Unverified Commit f7e2955f authored by Pavel Zwerschke's avatar Pavel Zwerschke
Browse files

Fix epochs as hyperparameter

parent e1c7db17
Branches hyperparam-optimization
Tags
1 merge request!3final submission for challenge
Pipeline #269255 passed
...@@ -442,9 +442,8 @@ def evaluate(params): ...@@ -442,9 +442,8 @@ def evaluate(params):
# In[22]: # In[22]:
# print(f'start fitting model in fold {fold_no}') # print(f'start fitting model in fold {fold_no}')
ep = 5
hist = cnn.fit(dg_train, hist = cnn.fit(dg_train,
epochs=ep, shuffle=False, epochs=epochs, shuffle=False,
validation_data=dg_valid) validation_data=dg_valid)
# In[23]: # In[23]:
...@@ -482,7 +481,7 @@ if __name__ == '__main__': ...@@ -482,7 +481,7 @@ if __name__ == '__main__':
space=space, space=space,
algo=tpe.suggest, algo=tpe.suggest,
trials=trials, trials=trials,
max_evals=2) max_evals=10)
print("---------------BEST---------------") print("---------------BEST---------------")
print(best) print(best)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment