Fix MSA plot

Signed-off-by: Antoni Baum <antoni.baum@protonmail.com>
pycaret · Mar 18, 2023 · 8e8e7ce · 8e8e7ce
1 parent cf3d0d0
commit 8e8e7ce
Showing 1 changed file with 11 additions and 7 deletions.
diff --git a/pycaret/internal/pycaret_experiment/supervised_experiment.py b/pycaret/internal/pycaret_experiment/supervised_experiment.py
@@ -4313,13 +4313,17 @@ def msa(show: bool = True):
             from interpret.blackbox import MorrisSensitivity
 
             try:
-                msa = MorrisSensitivity(
-                    predict_fn=model.predict_proba, data=test_X
-                )  # classification
-            except AttributeError:
-                msa = MorrisSensitivity(
-                    predict_fn=model.predict, data=test_X
-                )  # regression
+                # interpret>=0.3.1
+                msa = MorrisSensitivity(model=model, data=test_X)
+            except TypeError:
+                try:
+                    msa = MorrisSensitivity(
+                        predict_fn=model.predict_proba, data=test_X
+                    )  # classification
+                except AttributeError:
+                    msa = MorrisSensitivity(
+                        predict_fn=model.predict, data=test_X
+                    )  # regression
             msa_global = msa.explain_global()
             msa_plot = msa_global.visualize()
             if save: