Skip to content
Snippets Groups Projects

Resolve "BUG: mse monitoring plot not working with multiple branches"

1 file
+ 1
1
Compare changes
  • Side-by-side
  • Inline
@@ -223,7 +223,7 @@ class Training(RunEnvironment):
if multiple_branches_used:
filename = os.path.join(path, f"{name}_history_main_loss.pdf")
PlotModelHistory(filename=filename, history=history, main_branch=True)
if "mean_squared_error" in history.model.metrics_names:
if len([e for e in history.model.metrics_names if "mean_squared_error" in e]) > 0:
filename = os.path.join(path, f"{name}_history_main_mse.pdf")
PlotModelHistory(filename=filename, history=history, plot_metric="mse", main_branch=multiple_branches_used)
Loading