How to Run a Simple Machine Learning Model with scikit-learn

scikit-learn is a powerful and easy-to-use library for machine learning in Python. It provides simple and efficient tools for data analysis and machine learning. In this tutorial, we will walk through the steps of running a simple machine learning model using scikit-learn.

Prerequisites

• Python 3 installed on your system.
• Basic knowledge of Python programming.
• Familiarity with Jupyter Notebook or any Python development environment.

1. Installing Required Libraries

First, you need to install scikit-learn and other necessary libraries. Open your terminal or command prompt and run:

pip install scikit-learn numpy pandas matplotlib

2. Importing Libraries

Create a new Python file or Jupyter Notebook and import the necessary libraries:

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from sklearn.model_selection import train_test_split
from sklearn.linear_model import LinearRegression

3. Loading Data

For this example, we will use the classic Boston Housing dataset. You can load it directly from scikit-learn:

from sklearn.datasets import load_boston

boston_data = load_boston()
X = pd.DataFrame(boston_data.data, columns=boston_data.feature_names)
y = pd.Series(boston_data.target)

This code loads the dataset and separates the features (X) from the target variable (y).

4. Splitting the Data

Before training the model, split the data 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)

This will allocate 80% of the data for training and 20% for testing.

5. Training the Model

Now you can create a Linear Regression model and fit it to the training data:

model = LinearRegression()
model.fit(X_train, y_train)

6. Making Predictions

After training, you can use the model to make predictions on the test data:

predictions = model.predict(X_test)

7. Evaluating the Model

To evaluate the performance of the model, you can use metrics like Mean Absolute Error (MAE) or R-squared:

from sklearn.metrics import mean_absolute_error, r2_score

mae = mean_absolute_error(y_test, predictions)
r2 = r2_score(y_test, predictions)

print(f'MAE: {mae}')
print(f'R^2: {r2}') 

This will give you an idea of how well your model is performing.

8. Visualizing Predictions

You can visualize the model’s predictions versus actual values using a scatter plot:

plt.scatter(y_test, predictions)
plt.xlabel('Actual Values')
plt.ylabel('Predictions')
plt.title('Actual vs Predicted Values')
plt.show()

9. Conclusion

Congratulations! You have successfully built a simple machine learning model using scikit-learn. This tutorial covered loading data, training a model, making predictions, and evaluating its performance. As you advance, explore more complex models and parameters to enhance your machine learning projects!

About the Author

Dorian Kane

Author

View All Posts