Docker containerization in ML Python - ML Experiment: Train & Evaluate

Experiment - Docker containerization

Problem:You have a simple machine learning model trained with scikit-learn. You want to share it so others can run it easily without installing Python or libraries.

Current Metrics:Model trains with 90% accuracy on test data locally.

Issue:The model runs only on your machine. Others face setup issues due to different Python versions and missing packages.

Your Task

Create a Docker container that packages the model and its environment so anyone can run the model prediction with one command.

Use Python 3.12 base image.

Include all dependencies in the Dockerfile.

Provide a simple script inside the container to load the model and predict.

Hint 1

Hint 2

Hint 3

Hint 4

Solution

ML Python

import joblib
from sklearn.datasets import load_iris
from sklearn.ensemble import RandomForestClassifier

# Train and save model
iris = load_iris()
X, y = iris.data, iris.target
model = RandomForestClassifier(random_state=42)
model.fit(X, y)
joblib.dump(model, 'model.joblib')

# prediction.py
import joblib
import sys
import numpy as np

def main():
    model = joblib.load('model.joblib')
    # Example input: 5.1 3.5 1.4 0.2
    input_data = list(map(float, sys.argv[1:5]))
    prediction = model.predict([input_data])
    print(f'Predicted class: {prediction[0]}')

if __name__ == '__main__':
    main()

# Dockerfile
FROM python:3.12-slim
WORKDIR /app
COPY model.joblib ./
COPY prediction.py ./
RUN pip install scikit-learn joblib
ENTRYPOINT ["python", "prediction.py"]

Created a Python script 'prediction.py' to load the saved model and predict from command line input.

Saved the trained model as 'model.joblib' using joblib.

Wrote a Dockerfile using python:3.12-slim base image.

Installed required packages inside the container.

Copied model and script into the container.

Set the container entrypoint to run the prediction script.

Results Interpretation

Before: Model runs only on local machine with 90% accuracy but hard to share.

After: Model runs inside Docker container with same accuracy, easy to share and run anywhere.

Docker containerization packages your ML model and environment together, solving setup issues and making sharing and deployment easy and consistent.

Bonus Experiment

Extend the Docker container to serve the model via a simple web API using Flask.

💡 Hint

Add Flask to requirements, write a small Flask app to accept input and return prediction, expose port 5000 in Dockerfile.

Practice

1. What is the main benefit of using Docker for machine learning projects?

easy

A. It replaces the need for training data

B. It automatically improves model accuracy

C. It packages code and environment together for consistent runs

D. It speeds up GPU computations directly

Solution

1. Step 1: Understand Docker's purpose

   Docker bundles your ML code with all software and libraries needed.

2. Step 2: Identify the benefit for ML projects

   This bundling ensures your code runs the same on any machine, avoiding environment issues.

3. Final Answer:

   It packages code and environment together for consistent runs -> Option C

4. Quick Check:

   Docker = consistent environment [OK]

Hint: Docker bundles code + environment for consistency [OK]

Common Mistakes:

• Thinking Docker improves model accuracy
• Believing Docker replaces training data
• Assuming Docker speeds up hardware directly

2. Which of the following is the correct syntax to start a Docker container named ml_container from an image called ml_image?

easy

A. docker build ml_container ml_image

B. docker start ml_image --name ml_container

C. docker create ml_container ml_image

D. docker run --name ml_container ml_image

Solution

1. Step 1: Recall Docker run command usage

   The docker run command starts a container from an image and can assign a name with --name.

2. Step 2: Check each option

   docker run --name ml_container ml_image correctly uses docker run --name container image. Others misuse commands: start needs container ID, create creates but doesn't start, build builds images.

3. Final Answer:

   docker run --name ml_container ml_image -> Option D

4. Quick Check:

   docker run + --name = start container [OK]

Hint: Use 'docker run --name container image' to start named container [OK]

Common Mistakes:

• Using 'docker start' with image name instead of container
• Confusing 'docker create' with 'docker run'
• Using 'docker build' to start containers

3. Given this Dockerfile snippet for an ML project:

FROM python:3.8-slim
WORKDIR /app
COPY requirements.txt ./
RUN pip install -r requirements.txt
COPY . ./
CMD ["python", "train.py"]

What happens when you build and run this container?

medium

A. It only copies files but does not install dependencies

B. It installs dependencies and runs train.py inside the container

C. It runs train.py on the host machine, not inside container

D. It fails because CMD syntax is incorrect

Solution

1. Step 1: Analyze Dockerfile commands

   The Dockerfile uses FROM python:3.8-slim to get Python, sets working directory, copies requirements, installs them, copies code, then runs train.py.

2. Step 2: Understand container build and run

   Building creates an image with dependencies installed. Running executes python train.py inside the container environment.

3. Final Answer:

   It installs dependencies and runs train.py inside the container -> Option B

4. Quick Check:

   Dockerfile installs + runs script inside container [OK]

Hint: Dockerfile RUN installs, CMD runs script inside container [OK]

Common Mistakes:

• Thinking dependencies install on host, not container
• Believing CMD runs outside container
• Assuming CMD syntax is invalid here

4. You wrote a Dockerfile but when building the image, you get an error: ERROR: Could not open requirements.txt. What is the most likely cause?

medium

A. requirements.txt is not in the build context directory

B. Python version in FROM is too old

C. CMD syntax is wrong

D. Docker daemon is not running

Solution

1. Step 1: Understand build context and COPY command

   The COPY command copies files from the build context (folder where docker build runs). If requirements.txt is missing there, COPY fails.

2. Step 2: Check error cause

   Error says requirements.txt can't be opened, so likely it's not in the folder where build runs, causing failure.

3. Final Answer:

   requirements.txt is not in the build context directory -> Option A

4. Quick Check:

   Missing file in build context = COPY error [OK]

Hint: Ensure files are in build folder before docker build [OK]

Common Mistakes:

• Blaming Python version for file not found
• Assuming CMD syntax causes file errors
• Not checking if Docker daemon is running

5. You want to share your ML model training environment with a teammate. Which Docker approach ensures they get the exact same setup including Python version, libraries, and your training script?

hard

A. Create a Dockerfile specifying Python version, dependencies, and copy training code

B. Send them your Python script and ask them to install dependencies manually

C. Use a virtual environment on your machine and share it via email

D. Only share the trained model file without environment details

Solution

1. Step 1: Identify how to package environment and code

   A Dockerfile lets you specify exact Python version, libraries, and copies your code, creating a reproducible image.

2. Step 2: Compare options for sharing environment

   Manual installs or virtual envs are error-prone and not portable. Sharing only model misses environment needed for training.

3. Final Answer:

   Create a Dockerfile specifying Python version, dependencies, and copy training code -> Option A

4. Quick Check:

   Dockerfile = reproducible ML environment [OK]

Hint: Use Dockerfile to share full ML environment [OK]

Common Mistakes:

• Thinking manual installs guarantee same environment
• Believing virtualenv can be emailed and run anywhere
• Ignoring environment when sharing model files