DFIM test

ArnovanHilten · Nov 10, 2023 · ae83f6f · ae83f6f
1 parent f47b74e
commit ae83f6f
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 425 deletions.
diff --git a/GenNet.py b/GenNet.py
@@ -328,6 +328,12 @@ def make_parser_interpret(self, parser_topology):
             type=int,
             required=False,
             help='Select a layer for interpretation only necessary for NID')
+        parser_topology.add_argument(
+            '-num_eval',
+            type=int,
+            required=False,
+            default = 100,
+            help='Select a layer for interpretation only necessary for NID')
         return parser_topology
 
 

diff --git a/GenNet_utils/Interpret.py b/GenNet_utils/Interpret.py
@@ -90,7 +90,7 @@ def get_DFIM_scores(args):
     if args.genotype_path == "undefined":
         args.genotype_path = args.path
 
-    num_snps_to_eval = args.num_snps_to_eval if hasattr(args, 'num_snps_to_eval') else 100
+    num_snps_to_eval = args.num_eval if hasattr(args, 'num_eval') else 100
 
     model, masks = load_trained_network(args)
     part_n = 0  # placeholder solution for multiprocessing