Fix silly bug with feature selection

marcotcr · Dec 20, 2019 · 305b55b · 305b55b
1 parent 8b72193
commit 305b55b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lime/lime_base.py b/lime/lime_base.py
@@ -76,7 +76,7 @@ def feature_selection(self, data, labels, weights, num_features, method):
         elif method == 'forward_selection':
             return self.forward_selection(data, labels, weights, num_features)
         elif method == 'highest_weights':
-            clf = Ridge(alpha=0, fit_intercept=True,
+            clf = Ridge(alpha=0.01, fit_intercept=True,
                         random_state=self.random_state)
             clf.fit(data, labels, sample_weight=weights)