How to implement a random forest model in Python?
In the first example, we are going to implement a random forest model for a classification problem using the famous Iris dataset.
In the second example, we are going to implement a random forest model for a regression problem using the Boston housing dataset.
# Step 1: Import necessary libraries
from sklearn.ensemble import RandomForestClassifier
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score
# Step 2: Load the Iris dataset
iris = load_iris()
X = iris.data
y = iris.target
# Step 3: Split the dataset into training and testing sets
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
# Step 4: Create a Random Forest classifier and fit the model
rf = RandomForestClassifier(n_estimators=100)
rf.fit(X_train, y_train)
# Step 5: Make predictions on the test set
predictions = rf.predict(X_test)
# Step 6: Evaluate the model by calculating the accuracy
accuracy = accuracy_score(y_test, predictions)
print("Accuracy: ", accuracy)
In the second example, we are going to implement a random forest model for a regression problem using the Boston housing dataset.
# Step 1: Import necessary libraries
from sklearn.ensemble import RandomForestRegressor
from sklearn.datasets import load_boston
from sklearn.model_selection import train_test_split
from sklearn.metrics import mean_squared_error
# Step 2: Load the Boston housing dataset
boston = load_boston()
X = boston.data
y = boston.target
# Step 3: Split the dataset into training and testing sets
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
# Step 4: Create a Random Forest regressor and fit the model
rf = RandomForestRegressor(n_estimators=100)
rf.fit(X_train, y_train)
# Step 5: Make predictions on the test set
predictions = rf.predict(X_test)
# Step 6: Evaluate the model by calculating the mean squared error
mse = mean_squared_error(y_test, predictions)
print("Mean Squared Error: ", mse)
Comments
Post a Comment