Why advanced regression handles non-linearity in ML Python - Model Pipeline Impact

Model Pipeline - Why advanced regression handles non-linearity

This pipeline shows how advanced regression models learn to predict outcomes when the relationship between inputs and outputs is not a straight line. It transforms data, applies complex math, and improves predictions step by step.

Data Flow - 5 Stages

1Raw Data Input

1000 rows x 2 columns→Collect features and target values with non-linear relationship→1000 rows x 2 columns

Feature: temperature, Target: ice cream sales

↓

2Feature Engineering

1000 rows x 2 columns→Add polynomial features (square and cube of temperature)→1000 rows x 3 columns

Features: temperature, temperature^2, temperature^3, target

↓

3Train/Test Split

1000 rows x 3 columns→Split data into 800 training rows and 200 testing rows→Train: 800 rows x 3 columns, Test: 200 rows x 3 columns

Training data for model learning, testing data for evaluation

↓

4Model Training

800 rows x 3 feature columns→Fit polynomial regression model to training data→Trained model with learned coefficients

Model learns weights for temperature, temperature^2, temperature^3

↓

5Model Evaluation

200 rows x 3 feature columns→Predict target values and calculate error metrics→Predictions array of length 200, error metrics like RMSE

Compare predicted ice cream sales to actual sales

Training Trace - Epoch by Epoch

Loss
120.5 |**************
 45.3 |*******
 20.1 |****
 15.4 |***
      +----------------
       1  5 10 15 Epochs

Epoch	Loss ↓	Accuracy ↑	Observation
1	120.5	N/A	Initial model fit with high error
5	45.3	N/A	Model starts capturing non-linear pattern
10	20.1	N/A	Loss decreases steadily, model improves
15	15.4	N/A	Loss stabilizes, model fits data well

Prediction Trace - 3 Layers

Layer 1: Input Features

Layer 2: Polynomial Regression Model

Layer 3: Output Prediction

Model Quiz - 3 Questions

Why do we add squared and cubed features in advanced regression?

A To make the model run faster

B To reduce the number of features

C To capture curved relationships between features and target

D To remove noise from data

Key Insight

Advanced regression models handle non-linearity by creating new features that represent powers of original inputs. This lets the model learn curved patterns instead of just straight lines, improving prediction accuracy on complex data.

Practice

1. Why do advanced regression models handle non-linearity better than simple linear regression?

easy

A. Because they only use one feature at a time

B. Because they ignore data points that don't fit a line

C. Because they can model complex curved relationships in data

D. Because they always use fewer data points

Solution

1. Step 1: Understand simple linear regression limits

   Simple linear regression fits a straight line, so it cannot capture curves or bends in data.

2. Step 2: Recognize advanced regression capabilities

   Advanced regression models like decision trees or polynomial regression can fit curves and complex patterns.

3. Final Answer:

   Because they can model complex curved relationships in data -> Option C

4. Quick Check:

   Advanced regression models handle curves [OK]

Hint: Advanced regression fits curves, not just straight lines [OK]

Common Mistakes:

• Thinking advanced regression ignores data points
• Believing advanced regression uses fewer data points
• Assuming advanced regression only uses one feature

2. Which of the following is the correct way to create a polynomial regression model in Python using scikit-learn?

easy

A. from sklearn.preprocessing import PolynomialFeatures; poly = PolynomialFeatures(degree=2); X_poly = poly.fit_transform(X)

B. from sklearn.tree import DecisionTreeRegressor; model = DecisionTreeRegressor(); model.fit(X, y)

C. from sklearn.cluster import KMeans; model = KMeans(); model.fit(X)

D. from sklearn.linear_model import LinearRegression; model = LinearRegression(); model.fit(X_poly, y)

Solution

1. Step 1: Identify polynomial feature creation

   Polynomial regression requires transforming features using PolynomialFeatures to add powers of features.

2. Step 2: Recognize correct syntax for polynomial transformation

   from sklearn.preprocessing import PolynomialFeatures; poly = PolynomialFeatures(degree=2); X_poly = poly.fit_transform(X) correctly imports PolynomialFeatures and transforms X to X_poly for regression.

3. Final Answer:

   from sklearn.preprocessing import PolynomialFeatures; poly = PolynomialFeatures(degree=2); X_poly = poly.fit_transform(X) -> Option A

4. Quick Check:

   PolynomialFeatures creates polynomial features [OK]

Hint: Polynomial regression needs PolynomialFeatures to transform data [OK]

Common Mistakes:

• Confusing decision tree with polynomial regression
• Using clustering models for regression tasks
• Not transforming features before fitting polynomial regression

3. Given the code below, what will be the output of print(predictions)?

from sklearn.tree import DecisionTreeRegressor
X = [[1], [2], [3], [4], [5]]
y = [1, 4, 9, 16, 25]
model = DecisionTreeRegressor()
model.fit(X, y)
predictions = model.predict([[6]])
print(predictions)

medium

A. [16]

B. [36]

C. [9]

D. [25]

Solution

1. Step 1: Understand decision tree prediction behavior

   Decision trees predict by assigning the output of the closest training leaf node, not extrapolating beyond training data.

2. Step 2: Check training data and prediction input

   Input 6 is beyond training max 5, so prediction will be the leaf value for closest known input, which is 5 with output 25.

3. Final Answer:

   [25] -> Option D

4. Quick Check:

   Decision tree predicts closest leaf value = 25 [OK]

Hint: Decision trees do not extrapolate; predict closest known value [OK]

Common Mistakes:

• Assuming decision tree extrapolates like polynomial regression
• Expecting exact square of 6 (36) as output
• Confusing prediction with training labels

4. The following code tries to fit a polynomial regression but gives an error. What is the mistake?

from sklearn.linear_model import LinearRegression
from sklearn.preprocessing import PolynomialFeatures
X = [[1], [2], [3], [4]]
y = [1, 4, 9, 16]
model = LinearRegression()
X_poly = PolynomialFeatures(degree=2)
model.fit(X_poly, y)

medium

A. LinearRegression cannot fit polynomial data

B. X_poly is a class, not transformed data; need to call fit_transform on X

C. y should be a 2D array, not 1D

D. Degree should be 1 for polynomial features

Solution

1. Step 1: Identify how PolynomialFeatures is used

   PolynomialFeatures is a transformer class; it needs to be applied to X using fit_transform to create polynomial features.

2. Step 2: Spot the error in code

   Code assigns X_poly to the class instance, not the transformed data. The model.fit expects numeric array, not a class object.

3. Final Answer:

   X_poly is a class, not transformed data; need to call fit_transform on X -> Option B

4. Quick Check:

   Call fit_transform on X before fitting model [OK]

Hint: Call fit_transform on X before fitting model [OK]

Common Mistakes:

• Passing transformer class instead of transformed data
• Thinking LinearRegression can't fit polynomial features
• Misunderstanding y shape requirements

5. You have a dataset where the target variable changes in a complex curve with two features. Which approach best handles this non-linearity and why?

hard

A. Polynomial regression of degree 3 can model complex curves with multiple features

B. Simple linear regression will miss the curve

C. Decision tree with max depth 2 is too shallow to capture complexity

D. Dropping features reduces information and won't help non-linearity

Solution

1. Step 1: Analyze model capabilities for non-linearity

   Simple linear regression cannot model curves; decision tree with low depth may underfit; dropping features loses info.

2. Step 2: Evaluate polynomial regression for multiple features

   Polynomial regression with degree 3 creates interaction and power terms, capturing complex curves in multiple features.

3. Final Answer:

   Polynomial regression of degree 3 can model complex curves with multiple features -> Option A

4. Quick Check:

   Degree 3 polynomial regression models complex curves [OK]

Hint: Higher degree polynomial regression models complex curves well [OK]

Common Mistakes:

• Choosing shallow decision trees that underfit
• Dropping features reduces model power
• Using simple linear regression for curved data