Commit 1af9f3de authored by lucas_miranda's avatar lucas_miranda
Browse files

Added phenotype classification to train_model.py

parent 3113b526
Pipeline #89274 failed with stage
in 1 minute and 52 seconds
......@@ -776,4 +776,4 @@ class SEQ_2_SEQ_GMVAE:
# - Check usefulness of stateful sequential layers! (stateful=True in the LSTMs)
# - Investigate full covariance matrix approximation for the latent space! (details on tfp course) :)
# - Explore expanding the event dims of the final reconstruction layer
# - Gaussian Mixture as output layer? One component per bodypart (makes sense?)
\ No newline at end of file
# - Gaussian Mixture as output layer? One component per bodypart (makes sense?)
......@@ -405,18 +405,16 @@ if not tune:
description="weight applied to phenotypic classifier from the latent space",
)
)
metrics.append(
[
hp.Metric(
"phenotype_prediction_accuracy",
display_name="phenotype_prediction_accuracy",
),
hp.Metric(
"phenotype_prediction_auc",
display_name="phenotype_prediction_auc",
),
]
)
metrics += [
hp.Metric(
"phenotype_prediction_accuracy",
display_name="phenotype_prediction_accuracy",
),
hp.Metric(
"phenotype_prediction_auc",
display_name="phenotype_prediction_auc",
),
]
with tf.summary.create_file_writer(
os.path.join(output_path, "hparams", run_ID)
......
......@@ -120,4 +120,4 @@ rule explore_phenotype_classification:
"--window-size 11 "
"--window-step 11 "
"--stability-check 3 "
"--output-path {outpath}dimension_and_loss_experiments"
"--output-path {outpath}pheno_classification_experiments"
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