Commit 870f15a2 authored by lucas_miranda's avatar lucas_miranda
Browse files

Added extra branch to main autoencoder for rule_based prediction

parent deb4f0eb
Pipeline #98406 passed with stages
in 19 minutes and 54 seconds
......@@ -256,7 +256,7 @@ def tensorboard_metric_logging(
if phenotype_prediction:
idx = next(idx_generator)
pheno_acc = tf.keras.metrics.binary_accuracy(y_val[idx], outputs[idx])
pheno_acc = tf.keras.metrics.binary_accuracy(y_val[idx], tf.squeeze(outputs[idx]))
pheno_auc = tf.keras.metrics.AUC()
pheno_auc.update_state(y_val[idx], outputs[idx])
pheno_auc = pheno_auc.result().numpy()
......
Markdown is supported
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