📘

Bag of Words

Build a Bag-of-Words (BoW) representation for a corpus of documents.

Algorithm:

Build a vocabulary: collect all unique words across all documents, sorted alphabetically
For each document, create a vector where each element is the count of the corresponding vocabulary word in that document

Input format:

Output:

Input:

3
the cat sat
the dog sat
the cat and the dog

Output:

['and', 'cat', 'dog', 'sat', 'the']
[[0, 1, 0, 1, 1], [0, 0, 1, 1, 1], [1, 1, 1, 0, 2]]

Reasoning:

Step 1: Build vocabulary All unique words: {"the", "cat", "sat", "dog", "and"} Sorted: ["and", "cat", "dog", "sat", "the"]

Step 2: Count words per document

Editor

Python 3.13.1

0/0

Run code to see test results.