load from dir for plot_NN_2D script

scripts/plot_NN_2D.py

@@ -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")