Skip to content
Snippets Groups Projects
Commit 40c74808 authored by lukas leufen's avatar lukas leufen
Browse files

Merge branch...

Merge branch 'lukas_issue278_bug_mse-monitoring-plot-not-working-with-multiple-branches' into 'develop'

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

See merge request toar/mlair!243
parents 8105c732 90eb0b2e
No related branches found
No related tags found
4 merge requests!253include current develop,!252Resolve "release v1.3.0",!245update #275 branch,!243Resolve "BUG: mse monitoring plot not working with multiple branches"
Pipeline #60103 passed with warnings
......@@ -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)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment