Fixed attribute names issue

explainability/compute_shap_inter_vals.py

@@ -36,7 +36,7 @@ if __name__ == "__main__":
     # Setup
     # --------------------------------------------------------------------------------------------------------
     # Retrieve attribute names in order
-    attribute_names = list(np.load('../EDA/output/feature_names/feature_names.npy', allow_pickle=True))
+    attribute_names = list(np.load('../EDA/output/feature_names/all_features.npy', allow_pickle=True))
     # Reading data
     data_dic = read_test_data(attribute_names)
     method_names = {

explainability/compute_shap_vals.py

@@ -36,7 +36,7 @@ if __name__ == "__main__":
     # Setup
     # --------------------------------------------------------------------------------------------------------
     # Retrieve attribute names in order
-    attribute_names = list(np.load('../EDA/output/feature_names/feature_names.npy', allow_pickle=True))
+    attribute_names = list(np.load('../EDA/output/feature_names/all_features.npy', allow_pickle=True))
     # Reading data
     data_dic = read_test_data(attribute_names)
     method_names = {

explainability/fit_final_models.py

@@ -111,7 +111,7 @@ if __name__ == "__main__":
     # Setup
     # --------------------------------------------------------------------------------------------------------
     # Retrieve attribute names in order
-    attribute_names = list(np.load('../EDA/output/feature_names/feature_names.npy', allow_pickle=True))
+    attribute_names = list(np.load('../EDA/output/feature_names/all_features.npy', allow_pickle=True))
     # Reading data
     data_dic = read_training_data(attribute_names)
     method_names = {

explainability/shap_plots.ipynb

@@ -35,7 +35,7 @@
    "outputs": [],
    "source": [
     "# Retrieve attribute names in order\n",
-    "attribute_names = attribute_names = list(np.load('../EDA/output/feature_names/feature_names.npy', allow_pickle=True))\n",
+    "attribute_names = attribute_names = list(np.load('../EDA/output/feature_names/all_features.npy', allow_pickle=True))\n",
     "\n",
     "# Load test data\n",
     "X_test_pre = np.load('../gen_train_data/output/pre/X_test_pre.npy', allow_pickle=True)\n",
@@ -71,8 +71,8 @@
     "    \"UNDER\": \"XGB\"\n",
     "}\n",
     "\n",
-    "soc_var_names = np.load('../EDA/output/feature_names/soc_vars_names.npy', allow_pickle=True)\n",
-    "ind_var_names = np.load('../EDA/output/feature_names/ind_vars_names.npy', allow_pickle=True)"
+    "soc_var_names = np.load('../EDA/output/feature_names/social_factors.npy', allow_pickle=True)\n",
+    "ind_var_names = np.load('../EDA/output/feature_names/individual_factors.npy', allow_pickle=True)"
    ]
   },
   {