remove final additivity check

AIStream-Peelout · Jun 20, 2024 · 0889399 · 0889399
1 parent 98a37d5
commit 0889399
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/flood_forecast/evaluator.py b/flood_forecast/evaluator.py
@@ -143,7 +143,7 @@ def evaluate_model(
                 else:
                     end_tensor = test_data.inverse_scale(end_tensor.detach().reshape(-1, 1))
                 end_tensor_list = flatten_list_function(end_tensor.numpy().tolist())
-                end_tensor = end_tensor.squeeze(1)  # Removing extra dim from reshape?
+                end_tensor = end_tensor.squeeze(1)  # Removing extra dim from reshape.?
             history_length = model.params["dataset_params"]["forecast_history"]
             if "n_targets" in model.params:
                 df_train_and_test.loc[df_train_and_test.index[history_length:],

diff --git a/flood_forecast/explain_model_output.py b/flood_forecast/explain_model_output.py
@@ -146,8 +146,8 @@ def deep_explain_model_summary_plot(
         hist.cpu().numpy(), names=["batches", "observations", "features"]
     )
 
-    shap_values = deep_explainer.shap_values(history)
-    shap_values = fix_shap_values(shap_values, history)
+    shap_values = deep_explainer.shap_values(history, check_additivity=False)
+    shap_values = fix_shap_values(shap_values, history, check_additivity=False)
     shap_values = np.stack(shap_values)
     if len(shap_values.shape) != 4:
         shap_values = np.expand_dims(shap_values, axis=0)