DEV Community: Vaishnavi D

Creating a Docker Image and Pushing to DockerHub for a Python ML App

Vaishnavi D — Sat, 26 Apr 2025 04:56:16 +0000

Building and Deploying a Simple Machine Learning App with Docker

In this post, I’ll walk you through how I built a simple machine learning app using Python, containerized it with Docker, and published the image to DockerHub. This setup ensures the app can be run on any machine without worrying about dependencies or environment setup — super handy for portability and sharing!

The Machine Learning App

The app trains a logistic regression model using the Iris dataset to classify flowers based on sepal and petal measurements. Here's what it does step-by-step:

Loads the dataset (Iris.csv)
Displays basic info (shape + preview)
Generates and saves histograms and density plots
Splits the data for training and testing
Trains a logistic regression model using scikit-learn
Evaluates accuracy on the test set
Saves the model using Joblib

Make sure to keep Iris.csv in the same directory as app.py before running.

The Python Script (app.py)

from pandas import read_csv
from matplotlib import pyplot
from sklearn.linear_model import LogisticRegression
from sklearn.model_selection import train_test_split
import joblib

# Step 1: Load dataset
filename = "Iris.csv"
data = read_csv(filename)

# Step 2: Show data shape and preview
print("Shape of the dataset:", data.shape)
print("First 20 rows:\n", data.head(20))

# Step 3: Histograms
data.hist()
pyplot.savefig("histograms.png")
pyplot.close()

# Step 4: Density plots
data.plot(kind='density', subplots=True, layout=(3,3), sharex=False)
pyplot.savefig("density_plots.png")
pyplot.close()

# Step 5: Extract features and labels
array = data.values
X = array[:, 1:5]
Y = array[:, 5]

# Step 6: Train-test split
X_train, X_test, Y_train, Y_test = train_test_split(X, Y, test_size=0.33, random_state=7)

# Step 7: Train model
model = LogisticRegression(max_iter=200)
model.fit(X_train, Y_train)

# Step 8: Evaluate
result = model.score(X_test, Y_test)
print("Accuracy: {:.2f}%".format(result * 100))

# Step 9: Save the model
joblib.dump(model, "logistic_model.pkl")

Writing the Dockerfile

To containerize the app, I created a simple Dockerfile that:

Starts from a lightweight Python 3.9 image
Installs all required dependencies
Copies the app files into the container
Runs the script on startup

FROM python:3.9-slim

RUN pip install pandas scikit-learn matplotlib joblib

COPY . .

CMD ["python", "app.py"]

Building the Docker Image

With everything in place, I opened a terminal and ran:

docker build -t vaishnavi8754/24mcr121-ml:latest .

Docker pulled the base image, installed the dependencies, copied the files, and built the image successfully.

Pushing the Image to DockerHub

After logging into DockerHub using:

docker login

I pushed the image using:

docker push vaishnavi8754/24mcr121-ml:latest

Now the image is available at:

🔗 https://hub.docker.com/r/vaishnavi8754/24mcr121-ml

You (or anyone) can run it with:


docker pull vaishnavi8754/24mcr121-ml:latest
docker run vaishnavi8754/24mcr121-ml:latest

Advanced Git commands

Vaishnavi D — Fri, 25 Apr 2025 16:34:23 +0000

1. `git diff`

Shows changes between working directory and index or between commits.

git diff    

git diff --staged       
git diff HEAD

2. `git log`

Displays commit history.

git log       

git log --oneline       
git log --graph

3. `git clone`

Clones a repository from remote (e.g., GitHub) to your system.

git clone <repo_url>  


git clone https://github.com/username/project.git

4. `git pull`

Fetches and merges changes from the remote repository.

git pull origin main

5. `git push`

Pushes local commits to the remote repository.

git push origin main

6. `git blame`

Shows who made each line change and when.

git blame filename

7. `.gitignore`

Tells Git to ignore specific files/folders.

Create a .gitignore file in the root of your repo and add patterns:

node_modules/
.env
*.log
/dist

8. `git branch`

Manages branches in your repo.

git branch               
git branch new-feature   
git checkout new-feature

9. Git Merge Conflict

Occurs when merging changes and Git can't automatically resolve differences.

Steps:

git checkout main
git pull
git merge feature-branch  

git add .
git commit

10. Git Fork (on GitHub)

Forking creates a personal copy of someone else's repo.

Steps:

Go to a repo on GitHub.
Click Fork (top-right).
Clone your fork:

git clone https://github.com/your-username/forked-repo.git

11. GitHub Wiki

Used for project documentation inside the GitHub repo.

How to Use:

Go to your GitHub repository.
Click on the "Wiki" tab.
Click "Create the first page".
Add pages/sections with Markdown like .md files.

SCREENSHORTS:

Basic Git Commands

Vaishnavi D — Thu, 24 Apr 2025 03:30:06 +0000

git init

Initializes a new Git repository in your project directory. This sets up all necessary files and tracking to start version control.

    git init

git add

Adds changes in your working directory to the staging area. You can specify individual files or use . to add all changes.

            git add filename    
            git add .

git commit

Records the staged changes to the local repository with a message describing what was changed.

    git commit -m "Your commit message"

git push

Uploads your local repository content to a remote repository like GitHub.

    git push origin main

git status

Displays the state of the working directory and staging area. It shows which changes are staged, unstaged, or untracked.

    git status

git log

Shows the commit history for the repository, including commit IDs, messages, authors, and timestamps.

    git log

git branch

Lists all the branches in your repository. The * indicates the current branch.
git branch # List branches

git branch -M

Renames the current branch. Often used to rename master to main.

    git branch -M main        # Renames current branch to 'main'

git config user.name

Sets the username for Git commits. This is usually done globally.

    git config --global user.name "Your Name"

git config user.email

Sets the email address for Git commits. This should match your GitHub/GitLab account.

    git config --global user.email "you@example.com"

git remote add origin
Connects your local repository to a remote one. Replace with your actual repository URL.

    git remote add origin https://github.com/username/repository.git

DEV Community: Vaishnavi D

Creating a Docker Image and Pushing to DockerHub for a Python ML App

Building and Deploying a Simple Machine Learning App with Docker

The Machine Learning App

The Python Script (app.py)

Writing the Dockerfile

Building the Docker Image

Pushing the Image to DockerHub

Advanced Git commands

1. git diff

2. git log

3. git clone

4. git pull

5. git push

6. git blame

7. .gitignore

8. git branch

9. Git Merge Conflict

10. Git Fork (on GitHub)

11. GitHub Wiki

SCREENSHORTS:

Basic Git Commands

SCREENSHORTS:

1. `git diff`

2. `git log`

3. `git clone`

4. `git pull`

5. `git push`

6. `git blame`

7. `.gitignore`

8. `git branch`