update test

ArnovanHilten · Nov 4, 2024 · 3e5f720 · 3e5f720
1 parent 14daa72
commit 3e5f720
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/GenNet_utils/Dataloader.py b/GenNet_utils/Dataloader.py
@@ -275,7 +275,7 @@ def multi_genotype_matrix(self, idx):
     def get_data(self, sample_pat=0):
 
         genotype_hdf = tables.open_file(self.genotype_path + "/genotype.h5", "r")
-        ybatch = self.eval_subjects["labels"]
+
 
         if sample_pat > 0:
             self.eval_subjects = self.eval_subjects.sample(n=sample_pat, random_state=1)
@@ -286,8 +286,10 @@ def get_data(self, sample_pat=0):
         xcov = xcov.values
         xbatch = genotype_hdf.root.data[xbatchid,...]  
         xbatch = self.if_one_hot(xbatch)
+
+        ybatch = self.eval_subjects["labels"]
         ybatch = np.reshape(np.array(ybatch), (-1, 1))
-        ybatch = ybatch[xbatchid]
+
         genotype_hdf.close()
         return [xbatch, xcov], ybatch
 

diff --git a/interpretation/NID.py b/interpretation/NID.py
@@ -9,6 +9,7 @@
 import pandas as pd
 import numpy as np
 import itertools
+import math
 
 from scipy.sparse import coo_matrix