fix: 🐛 replace the hardcoded threshold by the user defined one, closes …

…#33
ThomasBury · Sep 22, 2023 · 14b33af · 14b33af
1 parent fddbc17
commit 14b33af
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/arfs/feature_selection/unsupervised.py b/src/arfs/feature_selection/unsupervised.py
@@ -446,7 +446,7 @@ def _recursive_collinear_elimination(association_matrix, threshold):
     dum = dum.drop(columns=most_collinear_feature, index=most_collinear_feature)
 
     while len(to_drop) > 1:
-        most_collinear_feature, to_drop = _most_collinear(dum, 0.75)
+        most_collinear_feature, to_drop = _most_collinear(dum, threshold)
         most_collinear_features.append(most_collinear_feature)
         dum = dum.drop(columns=most_collinear_feature, index=most_collinear_feature)
     return most_collinear_features