Setting Up a Simple ML Project: Tools and Platforms

By /

Welcome to the final blog in our “Introduction to Machine Learning and AI” series! If you’ve been following along, you now have a solid understanding of the basics of machine learning (ML) and artificial intelligence (AI). In this blog, we’ll dive into the practical side of things by setting up a simple ML project from scratch. We’ll explore the tools, platforms, and code you need to get started, and by the end, you’ll have a working ML model that you can tweak and expand upon.

This guide will walk you through a hands-on approach to setting up a simple ML project using popular tools, platforms, and frameworks.

🔧 Tools & Platforms for ML Development

Before diving into the code, let’s explore the essential tools and platforms you’ll need:

CategoryTools & PlatformsDescription
IDE / NotebookJupyter Notebook, VS Code, Google ColabInteractive coding environment for ML experiments.
ML FrameworksScikit-learn, TensorFlow, PyTorchLibraries for building and training models.
Data ProcessingPandas, NumPyHandling and preprocessing data.
VisualizationMatplotlib, Seaborn, PlotlyCreating charts and graphs for data insights.
Cloud PlatformsGoogle Cloud AI, AWS SageMaker, Azure MLCloud-based ML training and deployment.
Version ControlGit, GitHubKeeping track of code changes and collaboration.
Model DeploymentFlask, FastAPI, StreamlitAPIs for serving ML models.

1. Introduction to ML Project Setup

Before jumping into coding, it’s important to understand the workflow of a typical ML project:

  1. Problem Definition: What are you trying to solve?
  2. Data Collection: Gather the data needed for training.
  3. Data Preprocessing: Clean and prepare the data.
  4. Model Selection: Choose the right algorithm.
  5. Training: Train the model on your data.
  6. Evaluation: Test the model’s performance.
  7. Deployment: Use the model in real-world applications.

In this blog, we’ll focus on steps 2–6, using a real-world dataset and building a simple ML model.

2. Essential Tools and Platforms

To set up an ML project, you’ll need the following tools and platforms:

Programming Language: Python

Python is the most popular language for ML due to its simplicity and extensive libraries.

Libraries and Frameworks:

  • NumPy: For numerical computations.
  • Pandas: For data manipulation and analysis.
  • Scikit-learn: For ML algorithms and tools.
  • Matplotlib/Seaborn: For data visualization.
  • Jupyter Lab: An interactive environment for writing and running code.

Platforms:

  • Google Colab: A free cloud-based Jupyter notebook environment.
  • Kaggle: For datasets and competitions.
  • GitHub: For version control and collaboration.

3. Setting Up Your Environment

Let’s start by setting up your environment. If you’re using Jupyter Lab, make sure you have it installed. Alternatively, you can use Google Colab for a hassle-free setup.

Install Required Libraries

Run the following commands in your terminal or Jupyter notebook to install the necessary libraries:

pip install numpy pandas scikit-learn matplotlib seaborn jupyterlab

Verify Installation

Open a Python environment and run:

import numpy as np
import pandas as pd
import sklearn
import matplotlib.pyplot as plt
import seaborn as sns

print("All libraries installed successfully!")

4. Building a Simple ML Model

Let’s build a simple ML model to predict house prices using the Boston Housing Dataset. This dataset is included in Scikit-learn and is perfect for beginners.

Step 1: Load the Dataset

from sklearn.datasets import load_boston
import pandas as pd

# Load dataset
boston = load_boston()
data = pd.DataFrame(boston.data, columns=boston.feature_names)
data['PRICE'] = boston.target

# Display first 5 rows
print(data.head())

Step 2: Data Preprocessing

Split the data into features (X) and target (y):

X = data.drop('PRICE', axis=1)
y = data['PRICE']

Step 3: Split Data into Training and Testing Sets

from sklearn.model_selection import train_test_split

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

Step 4: Train a Model

Let’s use a simple Linear Regression model:

from sklearn.linear_model import LinearRegression

# Initialize the model
model = LinearRegression()

# Train the model
model.fit(X_train, y_train)

Step 5: Evaluate the Model

from sklearn.metrics import mean_squared_error, r2_score

# Make predictions
y_pred = model.predict(X_test)

# Evaluate
mse = mean_squared_error(y_test, y_pred)
r2 = r2_score(y_test, y_pred)

print(f"Mean Squared Error: {mse}")
print(f"R2 Score: {r2}")

5. Visualizing the Results

Visualization is key to understanding your model’s performance. Let’s plot the actual vs. predicted prices:

plt.figure(figsize=(10, 6))
plt.scatter(y_test, y_pred, alpha=0.7)
plt.plot([y_test.min(), y_test.max()], [y_test.min(), y_test.max()], 'k--', lw=2)
plt.xlabel('Actual Prices')
plt.ylabel('Predicted Prices')
plt.title('Actual vs Predicted House Prices')
plt.show()

Diagram: ML Workflow

Here’s a visual representation of the ML workflow:

Call To Action

Congratulations! You’ve just built your first ML model. But this is just the beginning. Here’s what you can do next:
Experiment with Other Algorithms: Try using Decision Trees, Random Forests, or Neural Networks.
Explore More Datasets: Check out datasets on Kaggle or UCI Machine Learning Repository.
Deploy Your Model: Learn how to deploy your model using Flask or FastAPI.
Join the Community: Share your projects on GitHub and collaborate with others.

Complete Code for a Simple ML Project in Google Colab

# Install Seaborn if not already installed
!pip install seaborn

# Import necessary libraries
import numpy as np
import pandas as pd
from sklearn.datasets import load_diabetes
from sklearn.model_selection import train_test_split
from sklearn.linear_model import LinearRegression
from sklearn.metrics import mean_squared_error, r2_score
import matplotlib.pyplot as plt
import seaborn as sns

# Step 1: Load the Diabetes Dataset
diabetes = load_diabetes()
data = pd.DataFrame(diabetes.data, columns=diabetes.feature_names)
data['PROGRESS'] = diabetes.target  # Target variable: disease progression

# Display the first 5 rows of the dataset
print("First 5 rows of the dataset:")
print(data.head())

# Step 2: Data Preprocessing
# Split the data into features (X) and target (y)
X = data.drop('PROGRESS', axis=1)
y = data['PROGRESS']

# Step 3: 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)

# Step 4: Train a Linear Regression Model
model = LinearRegression()
model.fit(X_train, y_train)

# Step 5: Make predictions on the test set
y_pred = model.predict(X_test)

# Step 6: Evaluate the model
mse = mean_squared_error(y_test, y_pred)
r2 = r2_score(y_test, y_pred)

print("\nModel Evaluation:")
print(f"Mean Squared Error: {mse}")
print(f"R2 Score: {r2}")

# Step 7: Visualize the results
plt.figure(figsize=(10, 6))
plt.scatter(y_test, y_pred, alpha=0.7, color='blue')
plt.plot([y_test.min(), y_test.max()], [y_test.min(), y_test.max()], 'k--', lw=2)
plt.xlabel('Actual Disease Progression')
plt.ylabel('Predicted Disease Progression')
plt.title('Actual vs Predicted Disease Progression')
plt.show()

Installation Output:

Requirement already satisfied: seaborn in /usr/local/lib/python3.11/dist-packages (0.13.2)
Requirement already satisfied: numpy!=1.24.0,>=1.20 in /usr/local/lib/python3.11/dist-packages (from seaborn) (1.26.4)
Requirement already satisfied: pandas>=1.2 in /usr/local/lib/python3.11/dist-packages (from seaborn) (2.2.2)
Requirement already satisfied: matplotlib!=3.6.1,>=3.4 in /usr/local/lib/python3.11/dist-packages (from seaborn) (3.10.0)
Requirement already satisfied: contourpy>=1.0.1 in /usr/local/lib/python3.11/dist-packages (from matplotlib!=3.6.1,>=3.4->seaborn) (1.3.1)
Requirement already satisfied: cycler>=0.10 in /usr/local/lib/python3.11/dist-packages (from matplotlib!=3.6.1,>=3.4->seaborn) (0.12.1)
Requirement already satisfied: fonttools>=4.22.0 in /usr/local/lib/python3.11/dist-packages (from matplotlib!=3.6.1,>=3.4->seaborn) (4.55.8)
Requirement already satisfied: kiwisolver>=1.3.1 in /usr/local/lib/python3.11/dist-packages (from matplotlib!=3.6.1,>=3.4->seaborn) (1.4.8)
Requirement already satisfied: packaging>=20.0 in /usr/local/lib/python3.11/dist-packages (from matplotlib!=3.6.1,>=3.4->seaborn) (24.2)
Requirement already satisfied: pillow>=8 in /usr/local/lib/python3.11/dist-packages (from matplotlib!=3.6.1,>=3.4->seaborn) (11.1.0)
Requirement already satisfied: pyparsing>=2.3.1 in /usr/local/lib/python3.11/dist-packages (from matplotlib!=3.6.1,>=3.4->seaborn) (3.2.1)
Requirement already satisfied: python-dateutil>=2.7 in /usr/local/lib/python3.11/dist-packages (from matplotlib!=3.6.1,>=3.4->seaborn) (2.8.2)
Requirement already satisfied: pytz>=2020.1 in /usr/local/lib/python3.11/dist-packages (from pandas>=1.2->seaborn) (2025.1)
Requirement already satisfied: tzdata>=2022.7 in /usr/local/lib/python3.11/dist-packages (from pandas>=1.2->seaborn) (2025.1)
Requirement already satisfied: six>=1.5 in /usr/local/lib/python3.11/dist-packages (from python-dateutil>=2.7->matplotlib!=3.6.1,>=3.4->seaborn) (1.17.0)

Dataset Preview:

First 5 rows of the dataset:
        age       sex       bmi        bp        s1        s2        s3  \
0  0.038076  0.050680  0.061696  0.021872 -0.044223 -0.034821 -0.043401   
1 -0.001882 -0.044642 -0.051474 -0.026328 -0.008449 -0.019163  0.074412   
2  0.085299  0.050680  0.044451 -0.005670 -0.045599 -0.034194 -0.032356   
3 -0.089063 -0.044642 -0.011595 -0.036656  0.012191  0.024991 -0.036038   
4  0.005383 -0.044642 -0.036385  0.021872  0.003935  0.015596  0.008142   
         s4        s5        s6  PROGRESS  
0 -0.002592  0.019907 -0.017646     151.0  
1 -0.039493 -0.068332 -0.092204      75.0  
2 -0.002592  0.002861 -0.025930     141.0  
3  0.034309  0.022688 -0.009362     206.0  
4 -0.002592 -0.031988 -0.046641     135.0

Model Evaluation:

Model Evaluation:
Mean Squared Error: 2900.193628493482
R2 Score: 0.4526027629719195

Visualization:
A scatter plot showing the actual vs. predicted disease progression values.

Final Thoughts

Setting up an ML project doesn’t have to be intimidating. With the right tools and platforms, you can quickly get started and build something meaningful. Remember, the key to mastering ML is practice, so keep experimenting and learning.

If you found this blog helpful, share it with your friends and colleagues. And don’t forget to leave a comment below with your thoughts or questions. Happy coding!

Call To Action

Share this blog with your network.
Try the code yourself and tweak it.
Subscribe to our newsletter for more hands-on tutorials.

By following this guide, you’ve taken a significant step toward becoming proficient in machine learning. Keep exploring, and soon you’ll be building complex models and solving real-world problems!

Leave a Comment

Your email address will not be published. Required fields are marked *

Scroll to Top