shaps

sgoldenlab · Dec 26, 2024 · 76973fd · 76973fd
1 parent 6882416
commit 76973fd
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/simba/mixins/train_model_mixin.py b/simba/mixins/train_model_mixin.py
@@ -1841,15 +1841,12 @@ def create_shap_log_mp(self,
                 check_int(name=f'{TrainModelMixin.create_shap_log_mp.__name__} save_file_no', value=save_file_suffix, min_value=0)
                 out_shap_path = os.path.join(save_dir, f"SHAP_values_{save_file_suffix}_{clf_name}.csv")
                 out_raw_path = os.path.join(save_dir, f"RAW_SHAP_feature_values_{save_file_suffix}_{clf_name}.csv")
-                df_save_paths = {
-                    'PRESENT': os.path.join(save_dir, f"SHAP_summary_{clf_name}_PRESENT_{save_file_suffix}.csv"),
-                    'ABSENT': os.path.join(save_dir, f"SHAP_summary_{clf_name}_ABSENT_{save_file_suffix}.csv")}
+                df_save_paths = {'PRESENT': os.path.join(save_dir, f"SHAP_summary_{clf_name}_PRESENT_{save_file_suffix}.csv"), 'ABSENT': os.path.join(save_dir, f"SHAP_summary_{clf_name}_ABSENT_{save_file_suffix}.csv")}
                 img_save_path = os.path.join(save_dir, f"SHAP_summary_line_graph_{clf_name}_{save_file_suffix}.png")
             else:
                 out_shap_path = os.path.join(save_dir, f"SHAP_values_{clf_name}.csv")
                 out_raw_path = os.path.join(save_dir, f"RAW_SHAP_feature_values_{clf_name}.csv")
-                df_save_paths = {'PRESENT': os.path.join(save_dir, f"SHAP_summary_{clf_name}_PRESENT.csv"),
-                                 'ABSENT': os.path.join(save_dir, f"SHAP_summary_{clf_name}_ABSENT.csv")}
+                df_save_paths = {'PRESENT': os.path.join(save_dir, f"SHAP_summary_{clf_name}_PRESENT.csv"), 'ABSENT': os.path.join(save_dir, f"SHAP_summary_{clf_name}_ABSENT.csv")}
                 img_save_path = os.path.join(save_dir, f"SHAP_summary_line_graph_{clf_name}.png")
 
             shap_df.to_csv(out_shap_path); raw_df.to_csv(out_raw_path)