update test

ArnovanHilten · Nov 4, 2024 · 14daa72 · 14daa72
1 parent 12ec6f1
commit 14daa72
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/GenNet_utils/Dataloader.py b/GenNet_utils/Dataloader.py
@@ -287,6 +287,7 @@ def get_data(self, sample_pat=0):
         xbatch = genotype_hdf.root.data[xbatchid,...]  
         xbatch = self.if_one_hot(xbatch)
         ybatch = np.reshape(np.array(ybatch), (-1, 1))
+        ybatch = ybatch[xbatchid]
         genotype_hdf.close()
         return [xbatch, xcov], ybatch