python - 如何确定 shap_values[0] 和 shap_values[1] 属于哪个类？

Question

据我了解，shap_values[0] 和 shap_values[1] 只是概率数组，如果 shap_values[0] 是 p，那么 shap_value[1] 是 1-p。但是，谁能告诉我如何验证 shap_values[0] 和 shap_values[1] 属于哪些类。

我的数据集中有两个类，但不知道哪个 shap_value 属于哪个类。

例如考虑我有类 - 错误，NO_ERROR

标签： pythonpandasshap

Answer 1

得到了答案 - 希望它对像我这样的初学者有所帮助。https://github.com/slundberg/shap/issues/1252#issuecomment-857469497

“您可以通过查看模型对象的 .classes_ 属性在 scikit-learn 模型中验证这一点。这将按顺序包含模型类的数组。”