Commit 335f267e authored by lucas_miranda's avatar lucas_miranda
Browse files

Changed default hyperparameter values

parent 71fe18c2
Pipeline #87083 failed with stage
in 32 minutes and 4 seconds
......@@ -56,8 +56,8 @@ def test_load_treatments():
def test_get_callbacks(
X_train, batch_size, variational, predictor, loss,
):
runID, tbc, cpc, cycle1c = deepof.train_utils.get_callbacks(
X_train, batch_size, variational, predictor, loss,
runID, tbc, cycle1c, cpc = deepof.train_utils.get_callbacks(
X_train, batch_size, True, variational, predictor, loss,
)
assert type(runID) == str
assert type(tbc) == tf.keras.callbacks.TensorBoard
......@@ -89,7 +89,7 @@ def test_tune_search(
):
callbacks = list(
deepof.train_utils.get_callbacks(
X_train, batch_size, hypermodel == "S2SGMVAE", predictor, loss,
X_train, batch_size, False, hypermodel == "S2SGMVAE", predictor, loss,
)
)[1:]
......
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