compute inception score in test steps

broadinstitute · Jan 20, 2025 · f064d9a · f064d9a
1 parent 12ed639
commit f064d9a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ml4h/plots.py b/ml4h/plots.py
@@ -471,7 +471,7 @@ def plot_metric_history(history, training_steps: int, title: str, prefix="./figu
         os.makedirs(os.path.dirname(figure_path))
     plt.savefig(figure_path)
     for log_label in ['loss', 'mse', 'mae', 'val_loss', 'val_mse', 'val_mae', 'n_loss', 'val_n_loss',
-                      'val_supervised_loss', 'val_kid']:
+                      'val_supervised_loss', 'val_kid', 'val_is']:
         if log_label not in history.history:
             continue
         logging.info(