Multi-class classification in ML Python - Model Pipeline Trace

Model Pipeline - Multi-class classification

This pipeline shows how a model learns to classify data into more than two categories. It starts with raw data, prepares it, trains a model, and then predicts the correct class for new data.

Data Flow - 6 Stages

1Raw data input

1500 rows x 4 columns→Collect dataset with features and labels for 3 classes→1500 rows x 4 columns

[5.1, 3.5, 1.4, 0.2, class=0]

↓

2Data preprocessing

1500 rows x 4 columns→Normalize feature values to range 0-1→1500 rows x 4 columns

[0.52, 0.70, 0.14, 0.10, class=0]

↓

3Train/test split

1500 rows x 4 columns→Split data into 1200 training and 300 testing rows→1200 rows x 4 columns (train), 300 rows x 4 columns (test)

[0.52, 0.70, 0.14, 0.10, class=0] (train)

↓

4Feature engineering

1200 rows x 4 columns→No new features added, use all 4 features→1200 rows x 4 columns

[0.52, 0.70, 0.14, 0.10, class=0]

↓

5Model training

1200 rows x 4 columns→Train neural network with softmax output for 3 classes→Model weights updated

Weights adjusted to reduce classification error

↓

6Model evaluation

300 rows x 4 columns→Predict classes and calculate accuracy→Accuracy score (e.g., 0.92)

Predicted classes vs true classes compared

Training Trace - Epoch by Epoch

Loss
1.1 |*       
0.9 | *      
0.7 |  *     
0.5 |   *    
0.3 |    **  
    +--------
     1 5 10 15 20 Epochs

Epoch	Loss ↓	Accuracy ↑	Observation
1	1.10	0.55	Model starts learning, loss high, accuracy low
5	0.75	0.75	Loss decreases, accuracy improves
10	0.50	0.85	Model converging, better predictions
15	0.35	0.90	Loss low, accuracy high, training stabilizes
20	0.30	0.92	Final epoch, model well trained

Prediction Trace - 4 Layers

Layer 1: Input layer

Layer 2: Hidden layer with ReLU activation

Layer 3: Output layer with softmax

Layer 4: Prediction

Model Quiz - 3 Questions

What does the softmax layer output represent?

A Probabilities for each class that sum to 1

B Raw scores that can be negative

C Binary yes/no decision

D Normalized input features

Key Insight

Multi-class classification uses a model that outputs probabilities for each class. Training reduces error over time, improving accuracy. Softmax ensures outputs are easy to interpret as class probabilities.

Practice

1. What does multi-class classification mean in machine learning?

easy

A. Sorting data into only two groups

B. Sorting data into three or more groups

C. Predicting continuous numbers

D. Clustering data without labels

Solution

1. Step 1: Understand classification types

   Binary classification sorts data into two groups, while multi-class sorts into three or more.

2. Step 2: Match definition to options

   Sorting data into three or more groups correctly states sorting into three or more groups, which matches multi-class classification.

3. Final Answer:

   Sorting data into three or more groups -> Option B

4. Quick Check:

   Multi-class = three or more groups [OK]

Hint: Multi-class means 3+ groups, not just 2 [OK]

Common Mistakes:

• Confusing multi-class with binary classification
• Thinking multi-class predicts numbers
• Mixing classification with clustering

2. Which of the following is the correct way to specify a multi-class classification model in Python using scikit-learn?

easy

A. from sklearn.linear_model import LogisticRegression\nmodel = LogisticRegression(multi_class='multinomial', solver='lbfgs')

B. from sklearn.linear_model import LogisticRegression\nmodel = LogisticRegression(multi_class='binary')

C. from sklearn.svm import SVC\nmodel = SVC(kernel='linear', multi_class=true)

D. from sklearn.tree import DecisionTreeClassifier\nmodel = DecisionTreeClassifier(multi_class='multinomial')

Solution

1. Step 1: Check scikit-learn multi-class syntax

   LogisticRegression supports multi_class='multinomial' with solver='lbfgs' for multi-class tasks.

2. Step 2: Evaluate each option

   from sklearn.linear_model import LogisticRegression\nmodel = LogisticRegression(multi_class='multinomial', solver='lbfgs') uses correct parameters. from sklearn.linear_model import LogisticRegression\nmodel = LogisticRegression(multi_class='binary') wrongly uses 'binary'. from sklearn.svm import SVC\nmodel = SVC(kernel='linear', multi_class=true)'s SVC does not have multi_class parameter. from sklearn.tree import DecisionTreeClassifier\nmodel = DecisionTreeClassifier(multi_class='multinomial')'s DecisionTreeClassifier does not accept multi_class parameter.

3. Final Answer:

   from sklearn.linear_model import LogisticRegression model = LogisticRegression(multi_class='multinomial', solver='lbfgs') -> Option A

4. Quick Check:

   LogisticRegression multi_class='multinomial' is correct [OK]

Hint: Use multi_class='multinomial' with LogisticRegression [OK]

Common Mistakes:

• Using multi_class='binary' for multi-class tasks
• Passing multi_class to models that don't accept it
• Forgetting to set solver='lbfgs' with multinomial

3. Given the following code, what will be the shape of the predicted output array?

from sklearn.datasets import load_iris
from sklearn.linear_model import LogisticRegression

iris = load_iris()
X, y = iris.data, iris.target
model = LogisticRegression(multi_class='multinomial', solver='lbfgs', max_iter=200)
model.fit(X, y)
predictions = model.predict(X)

medium

A. (3, 150)

B. (150, 3)

C. (150,)

D. (150, 1)

Solution

1. Step 1: Understand predict output shape

   For multi-class classification, predict returns a 1D array of class labels, one per sample.

2. Step 2: Check input data size

   iris dataset has 150 samples, so predictions shape is (150,)

3. Final Answer:

   (150,) -> Option C

4. Quick Check:

   Predict output shape = (number of samples,) [OK]

Hint: Predict returns 1D array of labels, length = number of samples [OK]

Common Mistakes:

• Expecting predict to return probabilities shape
• Confusing predict with predict_proba output
• Assuming output is 2D array always

4. You trained a multi-class classifier but it throws this error: ValueError: Unknown label type: 'continuous'. What is the most likely cause?

medium

A. The training data has too few samples

B. The model does not support multi-class classification

C. The input features have missing values

D. The target labels are continuous numbers instead of discrete classes

Solution

1. Step 1: Analyze error message

   ValueError about 'continuous' label type means labels are not discrete classes but continuous numbers.

2. Step 2: Match cause to options

   The target labels are continuous numbers instead of discrete classes correctly identifies continuous labels as cause. Other options do not relate to label type error.

3. Final Answer:

   The target labels are continuous numbers instead of discrete classes -> Option D

4. Quick Check:

   Continuous labels cause 'Unknown label type' error [OK]

Hint: Check if labels are discrete classes, not continuous numbers [OK]

Common Mistakes:

• Ignoring label type and focusing on features
• Assuming model limitation causes this error
• Not verifying label data format

5. You want to improve a multi-class classification model's performance on an imbalanced dataset with 5 classes. Which approach is best to try first?

hard

A. Use class weights to give more importance to minority classes during training

B. Reduce the number of classes to 2 by merging some classes

C. Increase the learning rate to speed up training

D. Remove samples from majority classes to balance dataset

Solution

1. Step 1: Understand imbalance problem

   Imbalanced classes cause model to favor majority classes, hurting minority class accuracy.

2. Step 2: Evaluate options for imbalance handling

   Using class weights (Use class weights to give more importance to minority classes during training) helps model focus on minority classes without losing data. Reducing classes (B) changes problem scope. Increasing learning rate (A) may harm training. Removing samples (D) loses valuable data.

3. Final Answer:

   Use class weights to give more importance to minority classes during training -> Option A

4. Quick Check:

   Class weights help handle imbalance best [OK]

Hint: Apply class weights to balance learning on imbalanced classes [OK]

Common Mistakes:

• Merging classes loses important distinctions
• Increasing learning rate can cause unstable training
• Removing data wastes valuable information