Skip to content
Snippets Groups Projects
Commit dce48709 authored by Dennis Noll's avatar Dennis Noll
Browse files

[keras] fixes bug in plotting (selection of ll vars)

parent 48739419
No related branches found
No related tags found
No related merge requests found
......@@ -635,7 +635,7 @@ class PlotMulticlass(PlottingCallback, TFSummaryCallback):
{
key: importance[key]
for key in importance.keys()
if any(key.endswith(p) for p in kinematics)
if any(key.endswith(f"_{p}") for p in kinematics)
},
xlabel="Importance [AU]",
)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment