update to the latest version

nnnnmj · Aug 21, 2022 · 0a80f94 · 0a80f94
1 parent 1b11a2a
commit 0a80f94
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/baseline/PyOD.py b/baseline/PyOD.py
@@ -103,7 +103,7 @@ def grid_search(self, X_train, y_train, ratio=None):
             # index of normal ana abnormal samples
             idx_a = np.where(y_train==1)[0]
             idx_n = np.where(y_train==0)[0]
-            idx_n = np.random.choice(idx_n, int(len(idx_a) / ratio), replace=True)
+            idx_n = np.random.choice(idx_n, int((len(idx_a) * (1-ratio)) / ratio), replace=True)
 
             idx = np.append(idx_n, idx_a) #combine
             np.random.shuffle(idx) #shuffle