First ML prediction (linear regression) in ML Python - Model Pipeline Trace

Model Pipeline - First ML prediction (linear regression)

This pipeline shows how a simple linear regression model learns to predict a number based on one input feature. It starts with data, prepares it, trains the model by finding the best line, and then uses that line to make predictions.

Data Flow - 6 Stages

1Data in

100 rows x 2 columns→Raw data with one feature and one target value→100 rows x 2 columns

Feature: 3.5, Target: 7.0

↓

2Preprocessing

100 rows x 2 columns→Split data into input feature and target arrays→Feature: 100 rows x 1 column, Target: 100 rows x 1 column

Feature: [3.5], Target: [7.0]

↓

3Feature Engineering

100 rows x 1 column→No additional features added, use feature as is→100 rows x 1 column

Feature: [3.5]

↓

4Model Trains

Feature: 100 rows x 1 column, Target: 100 rows x 1 column→Fit linear regression to find best slope and intercept→Model parameters: slope and intercept

Slope: 2.0, Intercept: 0.5

↓

5Metrics Improve

Training data predictions→Calculate loss (mean squared error) and accuracy (R^2 score)→Loss and accuracy values

Loss: 1.2, Accuracy: 0.85

↓

6Prediction

New feature input: 4.0→Apply model: prediction = slope * input + intercept→Predicted target value

Prediction: 8.5

Training Trace - Epoch by Epoch

Loss
10.0 |**********
 5.5 |******
 2.8 |****
 1.5 |**
 1.2 |**
     +----------------
      1  2  3  4  5 Epochs

Epoch	Loss ↓	Accuracy ↑	Observation
1	10.0	0.20	Model starts with high loss and low accuracy
2	5.5	0.50	Loss decreases, accuracy improves as model learns
3	2.8	0.75	Model fits data better, loss drops further
4	1.5	0.85	Good fit, loss low and accuracy high
5	1.2	0.88	Model converges, small improvements

Prediction Trace - 3 Layers

Layer 1: Input feature

Layer 2: Linear function

Layer 3: Output prediction

Model Quiz - 3 Questions

What does the model learn during training?

A The best slope and intercept for the line

B How to classify images

C How to cluster data points

D How to generate new data

Key Insight

Linear regression learns a simple line that best fits the data by minimizing the difference between predicted and actual values. As training continues, the model improves by reducing loss and increasing accuracy, enabling it to make good predictions on new inputs.

Practice

1. What is the main goal of linear regression in machine learning?

easy

A. To classify data into categories

B. To find a line that best fits the data points

C. To cluster data points into groups

D. To reduce the number of features

Solution

1. Step 1: Understand linear regression purpose

   Linear regression aims to find a straight line that best fits the relationship between input and output data.

2. Step 2: Compare with other options

   Classification, clustering, and feature reduction are different tasks, not the goal of linear regression.

3. Final Answer:

   To find a line that best fits the data points -> Option B

4. Quick Check:

   Linear regression = line fitting [OK]

Hint: Linear regression fits a line, not groups or categories [OK]

Common Mistakes:

• Confusing regression with classification
• Thinking linear regression clusters data
• Believing it reduces features

2. Which of the following is the correct Python syntax to create a simple linear regression model using scikit-learn?

easy

A. model = LinearRegression()

B. model = LinearRegression.fit()

C. model = LinearRegression.fit(X, y)

D. model = LinearRegression(X, y)

Solution

1. Step 1: Recall model creation syntax

   In scikit-learn, you first create the model object with LinearRegression(), without arguments.

2. Step 2: Understand fitting method

   The fit method is called on the model object with data, e.g., model.fit(X, y), not during creation.

3. Final Answer:

   model = LinearRegression() -> Option A

4. Quick Check:

   Model creation uses class name with parentheses [OK]

Hint: Create model with class(), fit separately [OK]

Common Mistakes:

• Calling fit during model creation
• Passing data to constructor
• Using fit as a class method

3. Given the code below, what will be the predicted value for X_new = [[5]]?

from sklearn.linear_model import LinearRegression
X = [[1], [2], [3], [4]]
y = [2, 4, 6, 8]
model = LinearRegression()
model.fit(X, y)
prediction = model.predict([[5]])
print(prediction)

medium

A. [9.]

B. [12.]

C. [8.]

D. [10.]

Solution

1. Step 1: Understand the training data pattern

   The output y doubles the input X: y = 2 * X.

2. Step 2: Predict for X_new = 5

   Using the pattern, prediction = 2 * 5 = 10.

3. Final Answer:

   [10.] -> Option D

4. Quick Check:

   2 x 5 = 10 [OK]

Hint: Check pattern in training data, apply to new input [OK]

Common Mistakes:

• Ignoring the pattern and guessing
• Using last y value as prediction
• Confusing input and output

4. What is wrong with the following code snippet for training a linear regression model?

from sklearn.linear_model import LinearRegression
X = [1, 2, 3, 4]
y = [2, 4, 6, 8]
model = LinearRegression()
model.fit(X, y)

medium

A. X should be a 2D array, not a 1D list

B. y should be a 2D array, not a list

C. LinearRegression() requires parameters in constructor

D. fit() method should be called with no arguments

Solution

1. Step 1: Check input shape requirements

   scikit-learn expects X as a 2D array (list of lists), but X is a 1D list here.

2. Step 2: Confirm y shape and method usage

   y can be 1D list, constructor needs no parameters, fit requires X and y arguments.

3. Final Answer:

   X should be a 2D array, not a 1D list -> Option A

4. Quick Check:

   X must be 2D for fit() [OK]

Hint: X must be 2D array for sklearn fit() [OK]

Common Mistakes:

• Passing 1D list for X
• Thinking y must be 2D
• Adding parameters to LinearRegression()

5. You trained a linear regression model on data where y = 3x + 2. After training, you want to predict the output for x = 10. Which code snippet correctly predicts and prints the result?

hard

A. model.predict([10])\nprint(prediction)

B. prediction = model.predict([10, 2])\nprint(prediction)

C. prediction = model.predict([[10]])\nprint(prediction[0])

D. prediction = model.predict(10)\nprint(prediction)

Solution

1. Step 1: Understand input shape for prediction

   scikit-learn expects a 2D array for prediction input, so [[10]] is correct.

2. Step 2: Access prediction output correctly

   predict returns an array; to print the single value, use prediction[0].

3. Final Answer:

   prediction = model.predict([[10]]) print(prediction[0]) -> Option C

4. Quick Check:

   Input 2D array, print first prediction [OK]

Hint: Use 2D input for predict, print first element [OK]

Common Mistakes:

• Passing 1D list or scalar to predict
• Printing prediction array directly without indexing
• Passing wrong shaped input with extra values