有没有办法在 Python 中为具有多个分类的随机森林制作部分依赖图（使用 scikit-learn）？

Posted 2023-03-12

【中文标题】有没有办法在 Python 中为具有多个分类的随机森林制作部分依赖图（使用 scikit-learn）？

【英文标题】：Is there a way to make partial dependence plots for random forest with multiple classification in Python (using scikit-learn)?

【发布时间】：2021-11-19 06:58:19

【问题描述】：

我对此提出了一个单独的问题，因为我不确定 scikit-learn 中是否存在这样的功能。我已经在 R 中看到了一些例子。如果该功能不存在，我会在 scikit-learn github 中提出请求，但只是想在提出请求之前与社区再次确认。

如果您知道除 scikit learn 之外的任何其他 Python 包可以执行该情节，请告诉我。谢谢。

【参考方案1】：

您必须指定要为其绘制部分依赖关系的类。这是由 plot_partial_dependence 函数中的参数“target”完成的

例如，如果您有三个目标类别“低”、“中”、“高”，您会说 plot_partial_dependence(estimator, ..., target='high')。

但是，我仍在尝试找到一些关于多类分类器的部分依赖图解释的答案。如果你有一些信息，请告诉我。