shap.py 406 Bytes
Newer Older
1 2 3 4 5 6 7 8 9 10
# Libraries
# --------------------------------------------------------------------------------------------------------
import shap
import numpy as np
# --------------------------------------------------------------------------------------------------------

# Load test data
X_test_pre = np.load('../gen_train_data/data/output/pre/X_test_pre.npy', allow_pickle=True)

print(list(X_test_pre.columns.values))