Skip to content

C
covid_analysis
Commit e541d5bf authored by Joaquin Torres's avatar Joaquin Torres
Browse files
Options

Attribute names now working

parent 9d601f43
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
explicability/shap_vals.py
View file @ e541d5bf
...@@ -136,8 +136,7 @@ if __name__ == "__main__": ...@@ -136,8 +136,7 @@ if __name__ == "__main__":
"UNDER": "XGB" "UNDER": "XGB"
} }
# Retrieve attribute names in order # Retrieve attribute names in order
df = pd.read_csv("../gen_train_data/data/input/pre_dataset.csv") attribute_names = list(np.load('../gen_train_data/data/output/attributes.npy', allow_pickle=True))
attribute_names = list(df.columns.values)
# -------------------------------------------------------------------------------------------------------- # --------------------------------------------------------------------------------------------------------
# Shap value generation # Shap value generation
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment