我目前正在使用SHAP库,我已经生成了每个特征的平均贡献图表,但我还想知道图表上显示的具体数值。
如何获取图表上显示的具体数值?
回答:
pd.DataFrame((zip(X.columns[np.argsort(np.abs(shap_values).mean(0))],np.abs(shap_values).mean(0))), columns=["feature", "importance" ]).sort_values(by="importance", ascending=False)
参考 GitHub