据我所知，sklearn已弃用partial dependence functionality。我尝试运行一个简单的示例:
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'对我来说，部分依赖性(和边际效应)对绘制一个非常重要的图(结合相对重要性)，以更好地理解机器学习结果和预测。
有没有其他选择？分别如何绘制部分依赖关系？
最佳答案
我认为sklearn的版本可能会造成混淆。只是一个建议-我会检查您的建议(例如import sklearn; sklearn.__version__)。例如，如果它是v.0.20.3，那么-您不是从partial_dependence而不是plot_partial_dependence寻找sklearn.ensemble.partial_dependence和sklearn.inspection吗？
关于python-3.x - 替代部分依赖图？，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/56189294/