From 2fdb3a329fb0a13a56838db55f10eb28a5eeac56 Mon Sep 17 00:00:00 2001 From: Sam Freesun Friedman Date: Tue, 4 Mar 2025 12:44:24 -0500 Subject: [PATCH] t2 random --- ml4h/plots.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ml4h/plots.py b/ml4h/plots.py index 433e231f3..39d40c59b 100755 --- a/ml4h/plots.py +++ b/ml4h/plots.py @@ -463,7 +463,7 @@ def plot_metric_history(history, training_steps: int, title: str, prefix="./figu col += 1 if col >= cols: break - if len(history.history[k]) < 1: + if len(history.history[k]) < 2: logging.info(f"Not enough epochs to plot learning curves at:{k}") return