From eed61151258ba85ffd80c94fcff0ac24fab36d9e Mon Sep 17 00:00:00 2001
From: Jackson Burns <33505528+JacksonBurns@users.noreply.github.com>
Date: Tue, 13 Feb 2024 21:03:37 -0500
Subject: [PATCH] add a TODO in shap

---
 fastprop/shap.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/fastprop/shap.py b/fastprop/shap.py
index fdd3c97..21bf3c9 100644
--- a/fastprop/shap.py
+++ b/fastprop/shap.py
@@ -92,6 +92,7 @@ def shap_fastprop(checkpoints_dir, input_file, importance_threshold=0.75):
             model_shap_values = [model_shap_values]
         per_model_shap.append(model_shap_values)
     all_shap_values = np.mean(np.array(per_model_shap), axis=0)
+    # TODO: write these to a file, sorted by highest absolute SHAP on first target.
 
     # for each target, create a plot of the most important features
     for i, target_name in enumerate(config_dict["targets"]):