Fix pdp plot

Signed-off-by: Antoni Baum <antoni.baum@protonmail.com>
pycaret · Mar 18, 2023 · bc80a3f · bc80a3f
1 parent 80df135
commit bc80a3f
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/pycaret/internal/pycaret_experiment/supervised_experiment.py b/pycaret/internal/pycaret_experiment/supervised_experiment.py
@@ -4285,13 +4285,17 @@ def pdp(show: bool = True):
             from interpret.blackbox import PartialDependence
 
             try:
-                pdp = PartialDependence(
-                    predict_fn=model.predict_proba, data=test_X
-                )  # classification
+                # interpret>=0.3.1
+                pdp = PartialDependence(model=model, data=test_X)
             except AttributeError:
-                pdp = PartialDependence(
-                    predict_fn=model.predict, data=test_X
-                )  # regression
+                try:
+                    pdp = PartialDependence(
+                        predict_fn=model.predict_proba, data=test_X
+                    )  # classification
+                except AttributeError:
+                    pdp = PartialDependence(
+                        predict_fn=model.predict, data=test_X
+                    )  # regression
 
             pdp_global = pdp.explain_global()
             pdp_plot = pdp_global.visualize(list(test_X.columns).index(pdp_feature))