Commit 259ae0ce authored by Manuel Guth's avatar Manuel Guth
Browse files

Merge branch update-puma with refs/heads/master into refs/merge-requests/572/train

parents 9edf1046 1724fe96
Pipeline #4135452 passed with stages
in 15 minutes and 56 seconds
......@@ -4,6 +4,7 @@
### Latest
- Switch to latest puma version (v0.1.3) [!572](https://gitlab.cern.ch/atlas-flavor-tagging-tools/algorithms/umami/-/merge_requests/572)
- Fixing docker image builds [!571](https://gitlab.cern.ch/atlas-flavor-tagging-tools/algorithms/umami/-/merge_requests/571)
- Fixing uncertainty calculation for the ROC curves [!566](https://gitlab.cern.ch/atlas-flavor-tagging-tools/algorithms/umami/-/merge_requests/566)
......
......@@ -2,7 +2,7 @@ matplotlib==3.5.1
mlxtend==0.19.0
numpy==1.21.0
pandas==1.3.5
puma-hep==0.1.1
puma-hep==0.1.3
pydash==5.1.0
ruamel.yaml==0.17.21
seaborn==0.11.2
......
......@@ -169,6 +169,7 @@ def plot_n_tracks_per_jet(
Histogram(
values=n_tracks_flavour,
flavour=flavour,
ratio_group=flavour,
label=label,
linestyle=linestyle,
),
......@@ -413,6 +414,7 @@ def plot_input_vars_trks(
Histogram(
values=track_values,
flavour=flavour,
ratio_group=flavour,
label=label,
linestyle=linestyle,
),
......@@ -556,6 +558,7 @@ def plot_input_vars_jets(
Histogram(
values=jets_flavour,
flavour=flavour,
ratio_group=flavour,
label=label,
linestyle=linestyle,
),
......
......@@ -688,6 +688,7 @@ def plot_score(
f"disc_{tagger}"
],
flavour=iter_flavour,
ratio_group=iter_flavour,
label=model_label if len(model_labels) > 1 else None,
linestyle=linestyle,
),
......@@ -780,6 +781,7 @@ def plot_prob(
f'{tagger}_{flav_cat[flavour]["prob_var_name"]}'
],
flavour=iter_flavour,
ratio_group=iter_flavour,
label=model_label if len(model_labels) > 1 else None,
linestyle=linestyle,
),
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment