diff --git a/ml4h/plots.py b/ml4h/plots.py index 85185ef81..fc1d7cc98 100755 --- 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(