updated/documented lrp and pytorch pipeline (#110)

* update plots * update conf matrix * lrp cleanup Former-commit-id: cf80664
jpata · May 6, 2022 · 9226111 · 9226111
1 parent d51439b
commit 9226111
Show file tree

Hide file tree

Showing 31 changed files with 2,287 additions and 3,385 deletions.
diff --git a/mlpf/lrp/__init__.py b/mlpf/lrp/__init__.py
@@ -1,8 +1,3 @@
-from lrp.args import parse_args
-from lrp.plots import make_heatmaps
-from lrp.model_io import model_io
-from lrp.model_lrp import PFNet7
-from lrp.gravnet_lrp import GravNetConv
-
-from lrp.lrp_clf_gpu import lrp_clf
-from lrp.lrp_reg_gpu import lrp_reg
+from lrp.lrp_mlpf import LRP_MLPF
+from lrp.model import MLPF
+from lrp.make_rmaps import make_Rmaps
diff --git a/mlpf/lrp/args.py b/mlpf/lrp/args.py
diff --git a/mlpf/lrp/gravnet_lrp.py b/mlpf/lrp/gravnet_lrp.py