  Courses

Explore our extensive collection of courses designed to help you master various subjects and skills. Whether you're a beginner or an advanced learner, there's something here for everyone.

Coding Bootcamp

Learn live

Upcoming live events

Learning library

For all the self-taught geeks out there, here is our content library with most of the learning materials we have produced throughout the years.

It makes sense to start learning by reading and watching videos about fundamentals and how things work.

Data Science and Machine Learning - 16 wks

Full-Stack Software Developer - 16w

Search from all Lessons

← Back to Lessons
Edit on Github
Open in Collab

# Exploring Random Forest

## Random forests in Python¶

Next we will see how we can implement this model in Python. To do so, we will use the scikit-learn library.

### Random forests for classification¶

To exemplify the implementation of a random forest for classification we will use the same data set as in the case of decision trees.

#### Step 1. Reading the processed dataset¶

In :
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split

X, y = load_iris(return_X_y = True, as_frame = True)

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size = 0.2, random_state = 42)


Out:
sepal length (cm)sepal width (cm)petal length (cm)petal width (cm)
224.63.61.00.2
155.74.41.50.4
656.73.14.41.4
114.83.41.60.2
424.43.21.30.2

The train set will be used to train the model, while the test will be used to evaluate its degree of effectiveness. Furthermore, it is not necessary for the predictor variables to be normalized, since random forests, and therefore decision trees, are not affected by the scale of the data because of the way they work: they make decisions based on certain feature thresholds, regardless of their scale.

#### Step 2: Initialization and training of the model¶

In :
from sklearn.ensemble import RandomForestClassifier

model = RandomForestClassifier(random_state = 42)
model.fit(X_train, y_train)

Out:
RandomForestClassifier(random_state=42)
In a Jupyter environment, please rerun this cell to show the HTML representation or trust the notebook.

Once the model has trained correctly, we can visualize the random forest with the same library. This visualization will show each complete derived tree:

In :
import matplotlib.pyplot as plt
from sklearn import tree

fig, axis = plt.subplots(2, 2, figsize = (15, 15))

# We show the first 4 trees out of the 100 generated (default)
tree.plot_tree(model.estimators_, ax = axis[0, 0], feature_names = list(X_train.columns), class_names = ["0", "1", "2"], filled = True)
tree.plot_tree(model.estimators_, ax = axis[0, 1], feature_names = list(X_train.columns), class_names = ["0", "1", "2"], filled = True)
tree.plot_tree(model.estimators_, ax = axis[1, 0], feature_names = list(X_train.columns), class_names = ["0", "1", "2"], filled = True)
tree.plot_tree(model.estimators_, ax = axis[1, 1], feature_names = list(X_train.columns), class_names = ["0", "1", "2"], filled = True)

plt.show() The training time of a model will depend, first of all, on the size of the dataset (instances and features), and also on the number of trees we want our random forest to have.

#### Step 3: Model prediction¶

Once the model has been trained, it can be used to predict with the test data set.

In :
y_pred = model.predict(X_test)
y_pred

Out:
array([1, 0, 2, 1, 1, 0, 1, 2, 1, 1, 2, 0, 0, 0, 0, 1, 2, 1, 1, 2, 0, 2,
0, 2, 2, 2, 2, 2, 0, 0])

With raw data it is very difficult to know whether the model is getting it right or not. To do this, we must compare it with reality. There are a large number of metrics to measure the effectiveness of a model in predicting, including accuracy, which is the fraction of predictions that the model made correctly.

In :
from sklearn.metrics import accuracy_score

accuracy_score(y_test, y_pred)

Out:
1.0

The model is perfect!

#### Step 4: Saving the model¶

Once we have the model we were looking for (presumably after hyperparameter optimization), to be able to use it in the future it is necessary to store it in our directory, together with the seed.

In :
from pickle import dump

dump(model, open("random_forest_classifier_default_42.sav", "wb"))


Adding an explanatory name to the model is vital, since in the case of losing the code that has generated it we will know, on the one hand, what configuration it has (in this case we say default because we have not customized any of the hyperparameters of the model, we have left those that the function has by default) and also the seed to replicate the random components of the model, which in this case we do it adding a number to the file name, the 42.

### Random forest for regression¶

To exemplify the implementation of a random forest we will use a data set with few instances and that has been previously treated with a full EDA. We will use the same data set as in the case of decision trees.

#### Step 1. Reading the processed dataset¶

In :
import pandas as pd


Out:
Petrol_taxAverage_incomePaved_HighwaysPopulation_Driver_licence(%)Petrol_Consumption
08.0444785770.529464
17.5487023510.529414
28.05319118680.451344
37.0434539050.672968
47.5335741210.547628
In :
X_train = train_data.drop(["Petrol_Consumption"], axis = 1)
y_train = train_data["Petrol_Consumption"]
X_test = test_data.drop(["Petrol_Consumption"], axis = 1)
y_test = test_data["Petrol_Consumption"]


The train set will be used to train the model, while the test will be used to evaluate its degree of effectiveness. Furthermore, it is not necessary for the predictor variables to be normalized, since random forests, and therefore decision trees, are not affected by the scale of the data because of the way they work: they make decisions based on certain feature thresholds, regardless of their scale.

#### Step 2: Initialization and training of the model¶

In :
from sklearn.ensemble import RandomForestRegressor

model = RandomForestRegressor(random_state = 42)
model.fit(X_train, y_train)

Out:
RandomForestRegressor(random_state=42)
In a Jupyter environment, please rerun this cell to show the HTML representation or trust the notebook.

#### Step 3: Model prediction¶

Once the model has been trained, it can be used to predict with the test sample of the dataset.

In :
y_pred = model.predict(X_test)
y_pred

Out:
array([598.62, 585.71, 581.46, 600.02, 497.24, 602.8 , 513.01, 831.44,
546.3 , 580.29])

To calculate the effectiveness of the model we will use the mean squared error (MSE):

In :
from sklearn.metrics import mean_squared_error

print(f"Mean squared error: {mean_squared_error(y_test, y_pred)}")

Mean squared error: 6835.456590000002


#### Step 4: Saving the model¶

Once we have the model we were looking for (presumably after hyperparameter optimization), to be able to use it in the future it is necessary to store it in our directory, together with the seed.

In :
dump(model, open("random_forest_regressor_default_42.sav", "wb"))


Adding an explanatory name to the model is vital, since in the case of losing the code that has generated it we will know, on the one hand, what configuration it has (in this case we say default because we have not customized any of the hyperparameters of the model, we have left those that the function has by default) and also the seed to replicate the random components of the model, which in this case we do it adding a number to the file name, the 42.