Commit 077e963d authored by lucas_miranda's avatar lucas_miranda
Browse files

Updated train_model.py to be compatible with phenotype classification

parent d207cd1d
Pipeline #86791 failed with stage
in 14 minutes and 57 seconds
......@@ -141,6 +141,9 @@ def tune_search(
X_train, y_train, X_val, y_val = data
if hypermodel == "S2SAE": # pragma: no cover
assert (
predictor == 0.0 and pheno_class == 0.0
), "Prediction branches are only available for variational models. See documentation for more details"
hypermodel = deepof.hypermodels.SEQ_2_SEQ_AE(input_shape=X_train.shape)
elif hypermodel == "S2SGMVAE":
......
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