如何在SHAP值ForcePlot中插入函数名称？

Question

我正在使用机器学习的SHAP值库来解释机器学习模型，当我使用ForcePlot时，它没有返回特征的名称，而是返回了该特征的具体值。

我需要更改哪个参数吗？（我之前使用时，它能正确显示特征的名称）

相关问题：

• 暂无相关问题

Answer 1

看起来你缺少了feature_names这个参数。你需要给它传入一个包含特征名称的列表，这样每个数值下面就会显示对应的特征名称。

如果你用的是数据框（dataframe），它会自动从列中推断出特征名称，但在你的情况下，似乎你用的是一个数组。