disable SHAP for NN

mljar · Jul 9, 2020 · 01e8639 · 01e8639
1 parent 495d82f
commit 01e8639
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/supervised/utils/shap.py b/supervised/utils/shap.py
@@ -66,9 +66,9 @@ def get_explainer(algorithm, X_train):
             explainer = shap.TreeExplainer(algorithm.model)
         elif algorithm.algorithm_short_name in ["Linear"]:
             explainer = shap.LinearExplainer(algorithm.model, X_train)
-        elif algorithm.algorithm_short_name in ["Neural Network"]:
-            explainer = shap.KernelExplainer(algorithm.model, X_train)
-
+        #elif algorithm.algorithm_short_name in ["Neural Network"]:
+        #    explainer = shap.GradientExplainer(algorithm.model, X_train) #, link="logit")
+        
         return explainer
 
     @staticmethod
@@ -186,6 +186,7 @@ def compute(
                 explainer = PlotSHAP.get_explainer(algorithm, X_train)
                 X_vald, y_vald = PlotSHAP.get_sample(X_validation, y_validation)
                 shap_values = explainer.shap_values(X_vald)
+
 
             # fix problem with 1 or 2 dimensions for binary classification
             expected_value = explainer.expected_value
@@ -245,6 +246,7 @@ def compute(
                     class_names,
                 )
         except Exception as e:
+            print(f"Exception while producing SHAP explanations. {str(e)}\nContinuing ...")
             logger.info(
                 f"Exception while producing SHAP explanations. {str(e)}\nContinuing ..."
             )