Cosine similarity in NLP - ML Experiment: Train & Evaluate

Experiment - Cosine similarity

Problem:You want to measure how similar two text sentences are using cosine similarity on their vector representations.

Current Metrics:Cosine similarity scores are computed but sometimes do not reflect true similarity because vectors are not normalized or text preprocessing is missing.

Issue:Cosine similarity values are inconsistent and sometimes low for clearly similar sentences due to lack of proper text vectorization and normalization.

Your Task

Improve cosine similarity calculation so that similar sentences have scores closer to 1 and dissimilar sentences have scores closer to 0.

Use only basic Python and sklearn libraries.

Do not use deep learning models or external APIs.

Keep the code simple and easy to understand.

Hint 1

Hint 2

Hint 3

Solution

NLP

from sklearn.feature_extraction.text import TfidfVectorizer
from sklearn.metrics.pairwise import cosine_similarity
import string

def preprocess(text):
    text = text.lower()
    return text.translate(str.maketrans('', '', string.punctuation))

sentences = [
    "I love machine learning.",
    "Machine learning is my passion!",
    "The sky is blue.",
    "I enjoy sunny days."
]

# Preprocess sentences
clean_sentences = [preprocess(s) for s in sentences]

# Convert sentences to TF-IDF vectors
vectorizer = TfidfVectorizer()
vectors = vectorizer.fit_transform(clean_sentences)

# Compute cosine similarity matrix
cosine_sim_matrix = cosine_similarity(vectors)

# Print similarity between first two sentences (expected high similarity)
print(f"Similarity between sentence 1 and 2: {cosine_sim_matrix[0,1]:.3f}")

# Print similarity between first and third sentence (expected low similarity)
print(f"Similarity between sentence 1 and 3: {cosine_sim_matrix[0,2]:.3f}")

Added text preprocessing to lowercase and remove punctuation.

Used TF-IDF vectorizer to convert text into normalized vectors.

Used sklearn's cosine_similarity function to compute similarity between vectors.

Results Interpretation

Before: Cosine similarity scores were inconsistent and sometimes low for similar sentences.
After: Similar sentences have scores around 0.7 indicating strong similarity, while dissimilar sentences have scores near 0.

Proper text preprocessing and vector normalization are essential for meaningful cosine similarity results in NLP tasks.

Bonus Experiment

Try using simple count vectorizer instead of TF-IDF and compare the cosine similarity scores.

💡 Hint

Replace TfidfVectorizer with CountVectorizer and observe how similarity values change.

Practice

1. What does cosine similarity measure between two vectors?

easy

A. The difference in vector lengths

B. How close the vectors point in the same direction

C. The sum of vector elements

D. The distance between vector origins

Solution

1. Step 1: Understand vector comparison

   Cosine similarity compares the angle between two vectors, not their length or sum.

2. Step 2: Interpret cosine similarity meaning

   A value close to 1 means vectors point in the same direction, showing similarity.

3. Final Answer:

   How close the vectors point in the same direction -> Option B

4. Quick Check:

   Cosine similarity = direction closeness [OK]

Hint: Cosine similarity checks angle, not length or sum [OK]

Common Mistakes:

• Confusing cosine similarity with Euclidean distance
• Thinking it measures vector length difference
• Assuming it sums vector values

2. Which of the following is the correct formula for cosine similarity between vectors A and B?

easy

A. \( \frac{\|A\|}{\|B\|} \)

B. \( \|A - B\| \)

C. \( \frac{A \cdot B}{\|A\| \times \|B\|} \)

D. \( A + B \)

Solution

1. Step 1: Recall cosine similarity formula

   Cosine similarity is the dot product of vectors divided by the product of their lengths.

2. Step 2: Match formula to options

   \( \frac{A \cdot B}{\|A\| \times \|B\|} \) matches the formula \( \frac{A \cdot B}{\|A\| \times \|B\|} \), others do not.

3. Final Answer:

   \( \frac{A \cdot B}{\|A\| \times \|B\|} \) -> Option C

4. Quick Check:

   Cosine similarity = dot product / product of norms [OK]

Hint: Look for dot product over product of lengths [OK]

Common Mistakes:

• Choosing Euclidean distance formula
• Adding vectors instead of dot product
• Dividing norms instead of multiplying

3. Given vectors A = [1, 2, 3] and B = [4, 5, 6], what is the cosine similarity (rounded to 2 decimals)?

medium

A. 0.97

B. 0.83

C. 0.74

D. 0.56

Solution

1. Step 1: Calculate dot product of A and B

   Dot product = 1*4 + 2*5 + 3*6 = 4 + 10 + 18 = 32

2. Step 2: Calculate norms of A and B

   Norm A = sqrt(1^2 + 2^2 + 3^2) = sqrt(14) ≈ 3.74; Norm B = sqrt(4^2 + 5^2 + 6^2) = sqrt(77) ≈ 8.77

3. Step 3: Compute cosine similarity

   Cosine similarity = 32 / (3.74 * 8.77) ≈ 32 / 32.83 ≈ 0.9749 rounded to 0.97

4. Step 4: Check closest option

   0.97 matches the value rounded to 2 decimals.

5. Final Answer:

   0.97 -> Option A

6. Quick Check:

   Dot product / (norms product) ≈ 0.97 [OK]

Hint: Calculate dot product and divide by product of lengths [OK]

Common Mistakes:

• Forgetting to take vector norms
• Mixing up dot product with element-wise multiplication
• Rounding too early causing wrong answer

4. What is wrong with this Python code to compute cosine similarity?

import numpy as np

def cosine_sim(a, b):
    return np.dot(a, b) / np.linalg.norm(a + b)

A = np.array([1, 0])
B = np.array([0, 1])
print(cosine_sim(A, B))

medium

A. It should add vectors before dot product

B. It uses np.dot instead of np.cross

C. It misses normalizing vectors before dot product

D. It divides by norm of sum instead of product of norms

Solution

1. Step 1: Analyze denominator in code

   The code divides by norm of (a + b), but cosine similarity requires product of norms of a and b.

2. Step 2: Understand correct formula

   Correct denominator is np.linalg.norm(a) * np.linalg.norm(b), not norm of sum.

3. Final Answer:

   It divides by norm of sum instead of product of norms -> Option D

4. Quick Check:

   Denominator must be product of norms [OK]

Hint: Denominator is product of norms, not norm of sum [OK]

Common Mistakes:

• Using norm of sum instead of product
• Confusing dot product with cross product
• Normalizing vectors before dot product unnecessarily

5. You have two text documents represented as TF-IDF vectors: doc1 = [0, 1, 2, 0] and doc2 = [1, 0, 1, 1]. Which step is best to improve cosine similarity comparison for very sparse vectors?

hard

A. Normalize vectors to unit length before computing cosine similarity

B. Add the vectors element-wise before similarity

C. Use Euclidean distance instead of cosine similarity

D. Ignore zero elements in vectors

Solution

1. Step 1: Understand sparse vector challenges

   Sparse vectors have many zeros; normalizing to unit length ensures fair angle comparison.

2. Step 2: Identify best practice for cosine similarity

   Normalizing vectors before cosine similarity avoids bias from vector length differences.

3. Final Answer:

   Normalize vectors to unit length before computing cosine similarity -> Option A

4. Quick Check:

   Normalization improves cosine similarity on sparse data [OK]

Hint: Always normalize vectors before cosine similarity [OK]

Common Mistakes:

• Adding vectors instead of comparing
• Switching to Euclidean distance without reason
• Ignoring zeros instead of normalizing