diff --git a/deepof/train_utils.py b/deepof/train_utils.py index 2ba52061856d399ad4b9a6bc52b6b1dd30bb9587..b874c11fc9923d6a877aed318b2c3295a5401592 100644 --- a/deepof/train_utils.py +++ b/deepof/train_utils.py @@ -113,7 +113,7 @@ def get_callbacks( elif reg_cat_clusters and reg_cluster_variance: latreg = "categorical+variance" - run_ID = "{}{}{}{}{}{}_{}".format( + run_ID = "{}{}{}{}{}{}{}_{}".format( ("GMVAE" if variational else "AE"), ("_Pred={}".format(predictor) if predictor > 0 and variational else ""), ("_Pheno={}".format(phenotype_class) if phenotype_class > 0 else ""),