chore: parallelize oversampling function

AFAgarap · AFAgarap · commit be40ccc81198 · 2021-02-04T23:05:36.000+08:00
diff --git a/pt_datasets/utils.py b/pt_datasets/utils.py
@@ -300,7 +300,7 @@ def oversample_dataset(
         The dataset that consists of the
         oversampled minority class with other classes.
     """
-    oversampler = SMOTE(random_state=seed)
+    oversampler = SMOTE(random_state=seed, n_jobs=-1)
     if len(features.shape) > 3:
         input_shape = features.shape
     if len(features.shape) > 2:
