Skip to content
Snippets Groups Projects
Commit e710a258 authored by Nikolai.Hartmann's avatar Nikolai.Hartmann
Browse files

load from dir for plot_NN_2D script

parent 0edb744b
No related branches found
No related tags found
No related merge requests found
......@@ -11,7 +11,7 @@ import ROOT
ROOT.gROOT.SetBatch()
ROOT.PyConfig.IgnoreCommandLineOptions = True
from KerasROOTClassification import ClassificationProject
from KerasROOTClassification import load_from_dir
from KerasROOTClassification.plotting import (
get_mean_event,
plot_NN_vs_var_2D,
......@@ -54,7 +54,7 @@ if args.all_neurons and (not args.mode.startswith("mean")):
if args.verbose:
logging.getLogger().setLevel(logging.DEBUG)
c = ClassificationProject(args.project_dir)
c = load_from_dir(args.project_dir)
plot_vs_activation = (args.vary == "activation")
......
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