add random seed to balancing method for consistent results

bagustris · Jul 1, 2024 · 09cd2e4 · 09cd2e4
1 parent 11f17ab
commit 09cd2e4
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/nkululeko/modelrunner.py b/nkululeko/modelrunner.py
@@ -238,21 +238,21 @@ def _check_balancing(self):
             if balancing == "ros":
                 from imblearn.over_sampling import RandomOverSampler
 
-                sampler = RandomOverSampler()
+                sampler = RandomOverSampler(random_state=42)
                 X_res, y_res = sampler.fit_resample(
                     self.feats_train, self.df_train[self.target]
                 )
             elif balancing == "smote":
                 from imblearn.over_sampling import SMOTE
 
-                sampler = SMOTE()
+                sampler = SMOTE(random_state=42)
                 X_res, y_res = sampler.fit_resample(
                     self.feats_train, self.df_train[self.target]
                 )
             elif balancing == "adasyn":
                 from imblearn.over_sampling import ADASYN
 
-                sampler = ADASYN()
+                sampler = ADASYN(random_state=42)
                 X_res, y_res = sampler.fit_resample(
                     self.feats_train, self.df_train[self.target]
                 )