Merge pull request #201 from gialmisi/master

Fixed skope rules not working with scikit-learn version 1.2 and up.
csinva · Feb 29, 2024 · a91641e · a91641e
2 parents 6efaa90 + 30a9a07
commit a91641e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/imodels/util/extract.py b/imodels/util/extract.py
@@ -109,7 +109,7 @@ def extract_skope(X, y, feature_names,
 
     for max_depth in max_depths:
         bagging_clf = BaggingRegressor(
-            base_estimator=DecisionTreeRegressor(
+            estimator=DecisionTreeRegressor(
                 max_depth=max_depth,
                 max_features=max_features,
                 min_samples_split=min_samples_split
@@ -192,4 +192,4 @@ def extract_marginal_curves(clf, X, max_evals=100):
         shape_function_vals = clf.predict_proba(dummy_input)[:, 1] - base
         feature_vals_list.append(feature_vals)
         shape_function_vals_list.append(shape_function_vals.tolist())
-    return feature_vals_list, shape_function_vals_list
+    return feature_vals_list, shape_function_vals_list