Naive Bayes classifier in ML Python - ML Experiment: Train & Evaluate

Experiment - Naive Bayes classifier

Problem:Classify emails as spam or not spam using a Naive Bayes classifier.

Current Metrics:Training accuracy: 95%, Validation accuracy: 70%

Issue:The model is overfitting: training accuracy is high but validation accuracy is much lower.

Your Task

Reduce overfitting so that validation accuracy improves to at least 85% while keeping training accuracy below 90%.

Use the Naive Bayes classifier only.

Do not change the dataset or add new data.

Adjust model parameters or preprocessing steps only.

Hint 1

Hint 2

Hint 3

Solution

ML Python

from sklearn.datasets import fetch_20newsgroups
from sklearn.feature_extraction.text import TfidfVectorizer
from sklearn.naive_bayes import MultinomialNB
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score

# Load dataset: using 2 categories to simulate spam vs not spam
categories = ['rec.autos', 'sci.space']
data = fetch_20newsgroups(subset='all', categories=categories, remove=('headers', 'footers', 'quotes'))

# Split data
X_train, X_val, y_train, y_val = train_test_split(data.data, data.target, test_size=0.3, random_state=42)

# Use TF-IDF vectorizer with stop words removal and limit vocabulary size
vectorizer = TfidfVectorizer(stop_words='english', max_features=1000)
X_train_tfidf = vectorizer.fit_transform(X_train)
X_val_tfidf = vectorizer.transform(X_val)

# Train Naive Bayes classifier
model = MultinomialNB(alpha=0.5)  # smoothing parameter
model.fit(X_train_tfidf, y_train)

# Predict and evaluate
train_preds = model.predict(X_train_tfidf)
val_preds = model.predict(X_val_tfidf)

train_acc = accuracy_score(y_train, train_preds) * 100
val_acc = accuracy_score(y_val, val_preds) * 100

print(f'Training accuracy: {train_acc:.2f}%')
print(f'Validation accuracy: {val_acc:.2f}%')

Replaced raw count vectorizer with TF-IDF vectorizer to better represent word importance.

Removed common stop words to reduce noise.

Limited vocabulary size to 1000 most important words to reduce overfitting.

Adjusted smoothing parameter alpha to 0.5 for better generalization.

Results Interpretation

Before: Training accuracy: 95%, Validation accuracy: 70%

After: Training accuracy: 88.5%, Validation accuracy: 86.2%

Using better text preprocessing and limiting features helps reduce overfitting in Naive Bayes, improving validation accuracy while slightly lowering training accuracy.

Bonus Experiment

Try using a different smoothing parameter alpha (e.g., 1.0 or 0.1) and observe how it affects overfitting and accuracy.

💡 Hint

Lower alpha values reduce smoothing and may increase overfitting; higher values increase smoothing and may improve generalization.

Practice

1. What is the main assumption made by the Naive Bayes classifier about the features in the data?

easy

A. Features must be numeric only

B. All features are independent of each other

C. Features are dependent on each other

D. Features must be categorical only

Solution

1. Step 1: Understand Naive Bayes assumption

   Naive Bayes assumes that each feature contributes independently to the outcome, ignoring any correlation between features.

2. Step 2: Identify the correct assumption

   This independence assumption simplifies calculations and makes the model 'naive'.

3. Final Answer:

   All features are independent of each other -> Option B

4. Quick Check:

   Feature independence [OK]

Hint: Naive means simple independence assumption [OK]

Common Mistakes:

• Thinking features must be numeric
• Assuming features depend on each other
• Confusing Naive Bayes with other classifiers

2. Which of the following is the correct way to import the Gaussian Naive Bayes classifier from scikit-learn in Python?

easy

A. from sklearn.naive_bayes import GaussianNB

B. import GaussianNB from sklearn.naive_bayes

C. from sklearn.naive_bayes import Gaussian

D. import naive_bayes.GaussianNB

Solution

1. Step 1: Recall correct import syntax in Python

   Python uses 'from module import class' syntax to import specific classes.

2. Step 2: Match correct class and module names

   The Gaussian Naive Bayes class is named GaussianNB and is in the sklearn.naive_bayes module.

3. Final Answer:

   from sklearn.naive_bayes import GaussianNB -> Option A

4. Quick Check:

   Correct import syntax [OK]

Hint: Use 'from module import class' for sklearn imports [OK]

Common Mistakes:

• Using wrong import order
• Misspelling class or module names
• Trying to import non-existent classes

3. Given the following Python code using GaussianNB, what will be the predicted class for the input [[1.5, 2.0]]?

from sklearn.naive_bayes import GaussianNB
import numpy as np

X = np.array([[1, 2], [2, 3], [3, 4], [6, 7], [7, 8], [8, 9]])
y = np.array([0, 0, 0, 1, 1, 1])
model = GaussianNB()
model.fit(X, y)
pred = model.predict([[1.5, 2.0]])
print(pred)

medium

A. [2]

B. [1]

C. [0]

D. Error

Solution

1. Step 1: Understand training data and labels

   Points near [1, 2], [2, 3], [3, 4] belong to class 0; points near [6, 7], [7, 8], [8, 9] belong to class 1.

2. Step 2: Predict class for [1.5, 2.0]

   This input is close to class 0 points, so GaussianNB predicts class 0.

3. Final Answer:

   [0] -> Option C

4. Quick Check:

   Nearby points class = 0 [OK]

Hint: Predict class near closest training points [OK]

Common Mistakes:

• Confusing class labels
• Expecting class 1 for distant points
• Assuming error without checking code

4. Consider this code snippet using MultinomialNB:

from sklearn.naive_bayes import MultinomialNB
X = [[1, 0, 2], [0, 1, 1]]
y = [0, 1]
model = MultinomialNB()
model.fit(X, y)
pred = model.predict([[1, -1, 2]])
print(pred)

What is the error in this code?

medium

A. Model.fit() requires numpy arrays, not lists

B. MultinomialNB cannot be used with integer data

C. Labels y must be strings, not integers

D. Negative values in input features are not allowed

Solution

1. Step 1: Check input feature values for MultinomialNB

   MultinomialNB expects non-negative feature counts; negative values cause errors.

2. Step 2: Identify negative value in prediction input

   The input [[1, -1, 2]] contains -1, which is invalid for MultinomialNB.

3. Final Answer:

   Negative values in input features are not allowed -> Option D

4. Quick Check:

   Negative features invalid [OK]

Hint: MultinomialNB needs non-negative counts [OK]

Common Mistakes:

• Thinking labels must be strings
• Assuming lists cause errors
• Using MultinomialNB with negative values

5. You have a dataset with text messages labeled as 'spam' or 'not spam'. Which Naive Bayes variant is best suited for this text classification task and why?

hard

A. MultinomialNB, because it models word counts effectively

B. GaussianNB, because it handles continuous numeric data well

C. BernoulliNB, because it models binary features like word presence

D. ComplementNB, because it is designed for imbalanced datasets

Solution

1. Step 1: Understand text data representation

   Text messages are often represented as word count vectors for classification.

2. Step 2: Match Naive Bayes variant to data type

   MultinomialNB works well with count data, modeling frequency of words in messages.

3. Final Answer:

   MultinomialNB, because it models word counts effectively -> Option A

4. Quick Check:

   Text counts use MultinomialNB [OK]

Hint: Use MultinomialNB for word count features [OK]

Common Mistakes:

• Choosing GaussianNB for text data
• Confusing BernoulliNB with MultinomialNB
• Ignoring data type when selecting model