Training data pipeline automation in MLOps - Mini Project: Build & Apply

Training Data Pipeline Automation

📖 Scenario: You are working as a machine learning engineer. You need to automate the process of preparing training data for your ML model. This involves collecting raw data, filtering it based on quality, and then outputting the cleaned data ready for training.

🎯 Goal: Build a simple Python script that automates a training data pipeline. The script will start with raw data, apply a quality filter, and then output the cleaned data.

📋 What You'll Learn

Create a dictionary with raw data samples and their quality scores

Add a quality threshold variable to filter data

Use a dictionary comprehension to select only data samples above the threshold

Print the filtered data dictionary

💡 Why This Matters

🌍 Real World

Automating data preparation saves time and reduces errors in machine learning projects by ensuring only good quality data is used for training.

💼 Career

Data engineers and ML engineers often build automated pipelines like this to prepare data efficiently and reliably for model training.

1

Create raw data dictionary

Create a dictionary called raw_data with these exact entries: 'sample1': 0.85, 'sample2': 0.45, 'sample3': 0.95, 'sample4': 0.30, 'sample5': 0.75 representing data sample names and their quality scores.

MLOps

# Create the raw_data dictionary with sample names and quality scores
# Your code here

Hint

Use curly braces to create a dictionary. Each entry has a sample name as a string key and a float value for quality.

2

Set quality threshold

Create a variable called quality_threshold and set it to 0.7 to filter out low-quality data samples.

MLOps

raw_data = {'sample1': 0.85, 'sample2': 0.45, 'sample3': 0.95, 'sample4': 0.30, 'sample5': 0.75}
# Create the quality_threshold variable and set it to 0.7
# Your code here

Hint

Just assign the number 0.7 to the variable named quality_threshold.

3

Filter data using dictionary comprehension

Create a new dictionary called filtered_data using dictionary comprehension. Include only those entries from raw_data where the quality score is greater than or equal to quality_threshold. Use sample and score as the loop variables.

MLOps

raw_data = {'sample1': 0.85, 'sample2': 0.45, 'sample3': 0.95, 'sample4': 0.30, 'sample5': 0.75}
quality_threshold = 0.7
# Create filtered_data dictionary with samples having score >= quality_threshold
# Your code here

Hint

Use dictionary comprehension syntax: {key: value for key, value in dict.items() if condition}.

4

Print filtered data

Write a print statement to display the filtered_data dictionary.

MLOps

raw_data = {'sample1': 0.85, 'sample2': 0.45, 'sample3': 0.95, 'sample4': 0.30, 'sample5': 0.75}
quality_threshold = 0.7
filtered_data = {sample: score for sample, score in raw_data.items() if score >= quality_threshold}
# Print the filtered_data dictionary
# Your code here

Hint

Use print(filtered_data) to show the filtered dictionary.

Practice

1. What is the main benefit of automating a training data pipeline in machine learning?

easy

A. It saves time and reduces human errors during data preparation.

B. It makes the model training faster by using GPUs.

C. It increases the size of the training dataset automatically.

D. It guarantees 100% accuracy of the machine learning model.

Solution

1. Step 1: Understand the purpose of automation in data pipelines

   Automation helps by handling repetitive tasks consistently without manual intervention.

2. Step 2: Identify the key benefits of automation

   Automation saves time and reduces errors that happen when humans prepare data manually.

3. Final Answer:

   It saves time and reduces human errors during data preparation. -> Option A

4. Quick Check:

   Automation = saves time and reduces errors [OK]

Hint: Automation mainly saves time and avoids mistakes [OK]

Common Mistakes:

• Thinking automation speeds up model training directly
• Assuming automation increases dataset size automatically
• Believing automation guarantees perfect model accuracy

2. Which of the following is the correct Python syntax to define a simple function that automates a data cleaning step?

easy

A. clean_data(data) => data.dropna()

B. def clean_data(data):\n return data.dropna()

C. def clean_data(data):\nreturn data.dropna()

D. function clean_data(data) { return data.dropna() }

Solution

1. Step 1: Identify correct Python function syntax

   Python functions start with 'def', followed by name and parameters, then indented body.

2. Step 2: Check indentation and syntax correctness

   def clean_data(data):\n return data.dropna() uses correct indentation and syntax; others use wrong language syntax or missing indentation.

3. Final Answer:

   def clean_data(data):\n return data.dropna() -> Option B

4. Quick Check:

   Python function syntax = def + indent + return [OK]

Hint: Python functions need 'def' and proper indentation [OK]

Common Mistakes:

• Using JavaScript syntax in Python
• Missing indentation after function definition
• Using arrow functions which are not Python syntax

3. Consider this Python code snippet automating a data pipeline step:

def normalize(data):
    mean = data.mean()
    std = data.std()
    return (data - mean) / std

import pandas as pd
sample = pd.Series([10, 20, 30])
result = normalize(sample)
print(result.round(2))

What is the printed output?

medium

A. [ -1.0, 0.0, 1.0 ]

B. [ -1.22, 0.00, 1.22 ]

C. [ 10, 20, 30 ]

D. [ 0.0, 0.0, 0.0 ]

Solution

1. Step 1: Calculate mean and standard deviation of the sample

   Mean = (10+20+30)/3 = 20; Std deviation = 10 (pandas std() uses ddof=1 by default).

2. Step 2: Normalize each value and round to 2 decimals

   (10-20)/10 = -1.0, (20-20)/10=0.0, (30-20)/10 = 1.0

3. Final Answer:

   [ -1.0, 0.0, 1.0 ] -> Option A

4. Quick Check:

   Normalization = (value-mean)/std [OK]

Hint: Normalize by subtracting mean and dividing by std [OK]

Common Mistakes:

• Confusing standard deviation with variance
• Not rounding output
• Returning original data instead of normalized

4. You have this code snippet for automating data loading:

def load_data(file_path):
    data = pd.read_csv(file_path)
    return data

# Usage
dataset = load_data('data.csv')
print(dataset.head())

But it throws an error: NameError: name 'pd' is not defined. How do you fix it?

medium

A. Remove the function and read CSV directly.

B. Change 'pd.read_csv' to 'csv.read'.

C. Add 'import pandas as pd' at the top of the script.

D. Rename 'file_path' to 'filepath' in the function.

Solution

1. Step 1: Understand the error message

   NameError means 'pd' is not recognized because pandas was not imported.

2. Step 2: Fix by importing pandas with alias 'pd'

   Add 'import pandas as pd' at the top so 'pd.read_csv' works correctly.

3. Final Answer:

   Add 'import pandas as pd' at the top of the script. -> Option C

4. Quick Check:

   Import pandas as pd to use pd.read_csv [OK]

Hint: Always import pandas as pd before using pd functions [OK]

Common Mistakes:

• Changing function parameter names without reason
• Assuming csv module replaces pandas read_csv
• Removing function instead of fixing import

5. You want to automate a training data pipeline that:
1. Loads CSV data,
2. Cleans missing values,
3. Normalizes numeric columns,
4. Saves the processed data.

Which tool or approach best supports scheduling and monitoring this pipeline automatically?

hard

A. Using Excel macros to clean and normalize data.

B. Writing a single Python script and running it manually each time.

C. Training the model directly without data preprocessing.

D. Using Apache Airflow to create and schedule pipeline tasks.

Solution

1. Step 1: Identify requirements for automation and monitoring

   We need a tool that schedules tasks and tracks their success or failure.

2. Step 2: Evaluate options for pipeline automation

   Apache Airflow is designed for scheduling, monitoring, and managing workflows automatically.

3. Final Answer:

   Using Apache Airflow to create and schedule pipeline tasks. -> Option D

4. Quick Check:

   Airflow = scheduling + monitoring pipelines [OK]

Hint: Use Airflow for automated scheduling and monitoring [OK]

Common Mistakes:

• Running scripts manually instead of automating
• Using Excel which lacks automation for pipelines
• Skipping data preprocessing before training