Commit 81cd01a5 authored by lucas_miranda's avatar lucas_miranda
Browse files

Updated train_model.py to be compatible with phenotype classification

parent cb8d01c0
......@@ -418,6 +418,7 @@ else:
k=k,
loss=loss,
overlap_loss=overlap_loss,
pheno_class=pheno_class,
predictor=predictor,
project_name="{}-based_{}_BAYESIAN_OPT".format(input_type, hyp),
callbacks=[tensorboard_callback, cp_callback, onecycle],
......
......@@ -104,6 +104,7 @@ def tune_search(
k: int,
loss: str,
overlap_loss: float,
pheno_class: float,
predictor: float,
project_name: str,
callbacks: List,
......@@ -122,6 +123,8 @@ def tune_search(
- loss (str): one of [ELBO, MMD, ELBO+MMD]
- overlap_loss (float): assigns as weight to an extra loss term which
penalizes overlap between GM components
- pheno_class (float): adds an extra regularizing neural network to the model,
which tries to predict the phenotype of the animal from which the sequence comes
- predictor (float): adds an extra regularizing neural network to the model,
which tries to predict the next frame from the current one
- project_name (str): ID of the current run
......
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