绘图sklearn RandomForestRegressor MSE

时间:2015-02-18 21:52:37

标签: python scikit-learn random-forest

有没有人知道如何用sklearn中的随机森林回归者绘制树木的MSE?

在R中,这非常容易:

> fit = randomForest(y ~ X)
> plot(fit)

但我还没有找到在python中做到这一点的方法。

1 个答案:

答案 0 :(得分:3)

像这样:

from sklearn.metrics import mean_squared_error

mean_squared_error(y, predictions)

或使用numpy:

np.mean((y_test - est.predict(X_test))**2)