In a previous article, we learned how to build a random forest classifier. The regressor improved our tree model by training many tree models and selecting the best. Now, we will see how to do the same for regression. In this article, we will see how to create a Random Forest Regressor in Sklearn.
To create a RandomForestRegressor, we use the RandomForestRegressor class from the ensemble module. We create a instance of RandomForestRegressor then pass our data to the fit method as we usually do when building models.
from sklearn.ensemble import RandomForestRegressor
from sklearn import datasets
iris = datasets.load_iris()
features = iris.data
target = iris.target
randomforest = RandomForestRegressor()
model = randomforest.fit(features, target)
print(model.score())