有没有人知道如何用sklearn中的随机森林回归者绘制树木的MSE?
在R中,这非常容易:
> fit = randomForest(y ~ X)
> plot(fit)
但我还没有找到在python中做到这一点的方法。
答案 0 :(得分:3)
像这样:
from sklearn.metrics import mean_squared_error
mean_squared_error(y, predictions)
或使用numpy:
np.mean((y_test - est.predict(X_test))**2)