Skip to content

Commit

Permalink
add a TODO in shap
Browse files Browse the repository at this point in the history
  • Loading branch information
JacksonBurns committed Feb 14, 2024
1 parent 2ad27e0 commit eed6115
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions fastprop/shap.py
Original file line number Diff line number Diff line change
Expand Up @@ -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"]):
Expand Down

0 comments on commit eed6115

Please sign in to comment.