Attribute names now working

explicability/shap_vals.py

@@ -136,8 +136,7 @@ if __name__ == "__main__":
         "UNDER": "XGB"
     }
     # Retrieve attribute names in order
-    df = pd.read_csv("../gen_train_data/data/input/pre_dataset.csv")
-    attribute_names = list(df.columns.values)
+    attribute_names = list(np.load('../gen_train_data/data/output/attributes.npy', allow_pickle=True))
     # --------------------------------------------------------------------------------------------------------
 
     # Shap value generation