Undo the random sample to som_utils

angelolab · Aug 16, 2021 · 89e6d50 · 89e6d50
1 parent b905e01
commit 89e6d50
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/ark/phenotyping/som_utils.py b/ark/phenotyping/som_utils.py
@@ -14,7 +14,6 @@
 from ark.utils import io_utils
 from ark.utils import load_utils
 from ark.utils import misc_utils
-from ark.utils import rand_utils
 
 
 def normalize_rows(pixel_data, channels):
@@ -308,9 +307,7 @@ def create_fov_pixel_data(fov, channels, img_data, seg_labels,
     pixel_mat['segmentation_label'] = seg_labels_flat
 
     # subset the pixel matrix for training
-    pixel_mat_subset = rand_utils.random_sample(
-        pixel_mat, subset_proportion=subset_proportion, seed=seed
-    )
+    pixel_mat_subset = pixel_mat.sample(frac=subset_proportion, random_state=seed)
 
     return pixel_mat, pixel_mat_subset