python-3.x - 替代部分依赖图？

Question

据我所知，sklearn已弃用部分依赖功能。我尝试运行一个简单的示例：

from sklearn.datasets import make_friedman1
from sklearn.ensemble import GradientBoostingRegressor
from sklearn.inspection import partial_dependence
from sklearn.inspection import plot_partial_dependence

X, y = make_friedman1()
clf = GradientBoostingRegressor(n_estimators=10).fit(X, y)
plot_partial_dependence(clf, X, [0, (0, 1)]) 

但我收到以下错误消息：ImportError: No module named 'sklearn.inspection'

对我来说，部分依赖（和边际效应）描绘了一个非常重要的（结合相对重要性）以更好地理解机器学习结果和预测。

有替代方案吗？分别如何绘制部分依赖？

Answer 1

我有同样的问题，我通过简单地更新 sklearn 解决了它，它现在包含 sklearn.inspection。我正在使用 Anaconda，如果您也在使用 Anaconda，只需在 Anaconda Propmt 中输入：

conda update --all

更新所有软件包。重新启动你的 jupyter notebook，现在它应该可以工作了。

Answer 2

我认为可能与sklearn. 就像一个建议——我会检查你的（例如，import sklearn; sklearn.__version__）。例如，如果它是v.0.20.3，碰巧 - 你不是在寻找partial_dependenceand plot_partial_dependencefromsklearn.ensemble.partial_dependence而不是sklearn.inspection吗？