Bag of Words (CountVectorizer) in NLP - Model Pipeline Trace

Model Pipeline - Bag of Words (CountVectorizer)

This pipeline converts text into numbers using the Bag of Words method. It counts how many times each word appears in the text. Then, a simple model learns to classify the text based on these counts.

Data Flow - 5 Stages

1Raw Text Input

5 samples (sentences)→Collect raw sentences as input data→5 samples (sentences)

["I love cats", "Cats are great pets", "Dogs are friendly", "I love dogs", "Pets are family"]

↓

2Text Preprocessing

5 samples (sentences)→Lowercase and remove punctuation→5 samples (cleaned sentences)

["i love cats", "cats are great pets", "dogs are friendly", "i love dogs", "pets are family"]

↓

3CountVectorizer (Bag of Words)

5 samples (cleaned sentences)→Convert sentences to word count vectors→5 samples x 8 features (unique words)

[[1,1,0,0,0,0,0,0], [0,1,1,1,1,0,0,0], [0,0,0,1,0,1,1,0], [1,1,0,0,0,0,1,0], [0,0,0,1,1,0,0,1]]

↓

4Train/Test Split

5 samples x 8 features→Split data into 4 training and 1 test samples→Training: 4 samples x 8 features, Test: 1 sample x 8 features

Training samples: 4 x 8, Test sample: 1 x 8

↓

5Model Training (Logistic Regression)

4 samples x 8 features→Train model to classify text based on word counts→Trained model

Model learns weights for each word feature

Training Trace - Epoch by Epoch

Loss
0.7 |****
0.6 |*** 
0.5 |**  
0.4 |**  
0.3 |*   
0.2 |*   
0.1 |    
    +------------
     1 2 3 4 5 Epochs

Epoch	Loss ↓	Accuracy ↑	Observation
1	0.65	0.50	Model starts with random guesses, accuracy is low
2	0.45	0.75	Model learns word importance, accuracy improves
3	0.30	0.85	Loss decreases steadily, model fits training data better
4	0.20	0.90	Model converges with high accuracy
5	0.15	0.95	Final epoch shows best performance

Prediction Trace - 4 Layers

Layer 1: Input Text

Layer 2: CountVectorizer

Layer 3: Model Prediction (Logistic Regression)

Layer 4: Final Decision

Model Quiz - 3 Questions

What does the CountVectorizer do to the input text?

A Counts how many times each word appears

B Translates text into another language

C Removes all vowels from the text

D Sorts words alphabetically

Key Insight

The Bag of Words method turns text into simple counts of words. This lets models learn patterns based on word frequency. As training progresses, the model improves by adjusting how much each word influences the prediction.

Practice

1. What does the Bag of Words model do in text processing?

easy

A. Counts how often each word appears in the text

B. Translates text into another language

C. Removes all punctuation from the text

D. Generates summaries of the text

Solution

1. Step 1: Understand Bag of Words purpose

   Bag of Words counts the frequency of each word in a text, ignoring order.

2. Step 2: Compare options to definition

   Only Counts how often each word appears in the text matches this description exactly.

3. Final Answer:

   Counts how often each word appears in the text -> Option A

4. Quick Check:

   Bag of Words = Counts words [OK]

Hint: Bag of Words counts words, not translates or summarizes [OK]

Common Mistakes:

• Confusing Bag of Words with translation
• Thinking it removes punctuation only
• Assuming it summarizes text

2. Which of the following is the correct way to import CountVectorizer from scikit-learn in Python?

easy

A. import CountVectorizer from sklearn.feature_extraction

B. from sklearn.feature_extraction.text import CountVectorizer

C. from sklearn.text import CountVectorizer

D. import CountVectorizer from sklearn.text

Solution

1. Step 1: Recall correct import path

   CountVectorizer is in sklearn.feature_extraction.text module.

2. Step 2: Match options to correct syntax

   Only from sklearn.feature_extraction.text import CountVectorizer uses the correct 'from ... import ...' syntax and correct module path.

3. Final Answer:

   from sklearn.feature_extraction.text import CountVectorizer -> Option B

4. Quick Check:

   Correct import path = from sklearn.feature_extraction.text import CountVectorizer [OK]

Hint: CountVectorizer is in sklearn.feature_extraction.text [OK]

Common Mistakes:

• Using wrong module path
• Incorrect import syntax
• Trying to import from sklearn.text

3. What will be the output shape of the matrix after applying CountVectorizer on these two sentences:
['I love cats', 'Cats love me']?

medium

A. (3, 2)

B. (2, 3)

C. (4, 2)

D. (2, 4)

Solution

1. Step 1: Identify unique words

   Words are: 'I', 'love', 'cats', 'me' (case insensitive, 'Cats' and 'cats' same).

2. Step 2: Count sentences and features

   There are 2 sentences and 4 unique words, so matrix shape is (2, 4).

3. Final Answer:

   (2, 4) -> Option D

4. Quick Check:

   2 sentences, 4 words = (2, 4) [OK]

Hint: Count unique words and sentences for shape (rows, columns) [OK]

Common Mistakes:

• Counting words per sentence instead of unique words
• Mixing rows and columns in shape
• Ignoring case sensitivity

4. The following code throws an error. What is the mistake?

from sklearn.feature_extraction.text import CountVectorizer
texts = ['hello world', 'hello']
vectorizer = CountVectorizer()
X = vectorizer.fit_transform(texts)
print(X.toarray())
print(vectorizer.get_feature_names())

medium

A. get_feature_names() is deprecated, should use get_feature_names_out()

B. fit_transform() should be fit_transform_text()

C. toarray() is not a method of X

D. CountVectorizer() needs a parameter for language

Solution

1. Step 1: Identify deprecated method

   get_feature_names() is deprecated in recent sklearn versions.

2. Step 2: Use correct method

   Replace get_feature_names() with get_feature_names_out() to fix error.

3. Final Answer:

   get_feature_names() is deprecated, should use get_feature_names_out() -> Option A

4. Quick Check:

   Use get_feature_names_out() not get_feature_names() [OK]

Hint: Use get_feature_names_out() instead of deprecated get_feature_names() [OK]

Common Mistakes:

• Thinking fit_transform() is wrong
• Assuming toarray() is invalid
• Believing CountVectorizer needs language parameter

5. You have a list of sentences with some words repeated many times. How can you use CountVectorizer to ignore words that appear in more than 50% of the sentences?

hard

A. Set min_df=0.5 to ignore frequent words

B. Use stop_words='english' to remove frequent words

C. Set the parameter max_df=0.5 when creating CountVectorizer

D. Set max_features=0.5 to limit word count

Solution

1. Step 1: Understand max_df parameter

   max_df=0.5 tells CountVectorizer to ignore words in more than 50% of documents.

2. Step 2: Compare other options

   min_df controls minimum frequency, stop_words removes common English words, max_features limits number of features, none ignore frequent words by percentage.

3. Final Answer:

   Set the parameter max_df=0.5 when creating CountVectorizer -> Option C

4. Quick Check:

   max_df filters frequent words by document frequency [OK]

Hint: Use max_df to exclude very common words [OK]

Common Mistakes:

• Confusing max_df with min_df
• Thinking stop_words removes all frequent words
• Using max_features to filter frequency