Merge pull request #18 from AC-BO-Hackathon:subsampling-function

Subsampling-function
AC-BO-Hackathon · Mar 27, 2024 · 87218b5 · 87218b5
2 parents 421de83 + ac61b72
commit 87218b5
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/utils/subsampling.py b/utils/subsampling.py
@@ -0,0 +1,7 @@
+import numpy as np
+
+def random_subsample(df, num_samples):
+    np.random.seed(42)
+    indices = np.random.choice(df.index, num_samples, replace=False)
+    subsampled_df = df.loc[indices]
+    return subsampled_df