Plot sklearn RandomForestRegressor MSE

Question

Кто-нибудь знает способ построения MSE деревьев из случайного лесного регрессора в sklearn?

В R это невероятно легко:

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

, но я не нашел способа сделать это в Python.

Answer 1

Как это:

from sklearn.metrics import mean_squared_error 

mean_squared_error(y, predictions) 

или с помощью NumPy:

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