Word embeddings concept (Word2Vec) in ML Python

Introduction

Word embeddings turn words into numbers so computers can understand language better. Word2Vec is a popular way to do this by learning word meanings from lots of text.

When you want to find similar words, like 'king' and 'queen'.

When building chatbots that understand user questions.

When analyzing text to find topics or emotions.

When improving search engines to find related words.

When creating recommendation systems based on text data.

Syntax

ML Python

from gensim.models import Word2Vec

# sentences is a list of tokenized sentences
model = Word2Vec(sentences, vector_size=100, window=5, min_count=1, workers=4)

# Get vector for a word
vector = model.wv['word']

# Find similar words
similar = model.wv.most_similar('word')

sentences must be a list of lists, where each inner list is a sentence split into words.

vector_size controls the length of the word vectors (usually 50-300).

Examples

This creates a Word2Vec model with smaller vectors and a smaller context window, ignoring words that appear less than twice.

ML Python

model = Word2Vec(sentences, vector_size=50, window=3, min_count=2)

Gets the vector (list of numbers) that represents the word 'apple'.

ML Python

vector = model.wv['apple']

Finds the top 3 words most similar to 'king' based on learned word vectors.

ML Python

similar_words = model.wv.most_similar('king', topn=3)

Sample Model

This code trains a Word2Vec model on a few simple sentences. It then shows the vector for the word 'learning' and finds the top 3 words that are most similar to 'learning' based on the model.

ML Python

from gensim.models import Word2Vec

# Sample sentences
sentences = [
    ['I', 'love', 'machine', 'learning'],
    ['machine', 'learning', 'is', 'fun'],
    ['I', 'enjoy', 'learning', 'new', 'things'],
    ['deep', 'learning', 'is', 'a', 'part', 'of', 'machine', 'learning']
]

# Train Word2Vec model
model = Word2Vec(sentences, vector_size=50, window=2, min_count=1, workers=1, seed=42)

# Get vector for 'learning'
vector = model.wv['learning']

# Find words similar to 'learning'
similar_words = model.wv.most_similar('learning', topn=3)

print(f"Vector for 'learning' (first 5 values): {vector[:5]}")
print("Top 3 words similar to 'learning':")
for word, score in similar_words:
    print(f"{word}: {score:.3f}")

Important Notes

Word2Vec learns word meanings by looking at nearby words in sentences.

Words with similar meanings end up with similar vectors.

More training data usually means better word vectors.

Summary

Word2Vec converts words into numbers that capture meaning.

It learns from the context words appear in.

These vectors help computers understand and compare words.

Practice

1. What is the main purpose of Word2Vec in machine learning?

easy

A. To translate text from one language to another

B. To count the number of words in a sentence

C. To convert words into numerical vectors that capture their meanings

D. To generate random words for testing

Solution

1. Step 1: Understand Word2Vec's role

   Word2Vec creates vectors representing words based on their context, capturing meaning.

2. Step 2: Compare options

   Only To convert words into numerical vectors that capture their meanings describes converting words into meaningful numerical vectors.

3. Final Answer:

   To convert words into numerical vectors that capture their meanings -> Option C

4. Quick Check:

   Word2Vec = word vectors [OK]

Hint: Word2Vec turns words into numbers showing meaning [OK]

Common Mistakes:

• Thinking Word2Vec translates languages
• Confusing Word2Vec with word counting
• Assuming Word2Vec generates random words

2. Which of the following is the correct way to import Word2Vec from the gensim library in Python?

easy

A. import gensim.Word2Vec

B. import Word2Vec from gensim.models

C. from gensim import Word2Vec

D. from gensim.models import Word2Vec

Solution

1. Step 1: Recall correct Python import syntax

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

2. Step 2: Match with gensim syntax

   The correct import is 'from gensim.models import Word2Vec'. Other options have wrong syntax.

3. Final Answer:

   from gensim.models import Word2Vec -> Option D

4. Quick Check:

   Correct import syntax = from gensim.models import Word2Vec [OK]

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

Common Mistakes:

• Using 'import Word2Vec from ...' which is invalid syntax
• Trying to import directly from 'gensim' without 'models'
• Using dot notation in import statements incorrectly

3. Given the following code snippet, what will be the output of print(model.wv.most_similar('king', topn=1))?

from gensim.models import Word2Vec
sentences = [['king', 'queen', 'man', 'woman'], ['prince', 'princess', 'boy', 'girl']]
model = Word2Vec(sentences, vector_size=10, window=2, min_count=1, epochs=10)

medium

A. [('man', similarity_score)] where similarity_score is close to 0

B. [('queen', similarity_score)] where similarity_score is close to 1

C. Error because 'king' is not in vocabulary

D. Empty list []

Solution

1. Step 1: Check if 'king' is in vocabulary

   'king' appears in the sentences, so it is in the model's vocabulary.

2. Step 2: Understand most_similar output

   Word2Vec finds words with similar context. 'queen' is closely related to 'king' in the data, so it will have highest similarity.

3. Final Answer:

   [('queen', similarity_score)] where similarity_score is close to 1 -> Option B

4. Quick Check:

   Most similar to 'king' = 'queen' [OK]

Hint: Most similar words share context, 'queen' pairs with 'king' [OK]

Common Mistakes:

• Assuming 'king' is missing from vocabulary
• Expecting unrelated words like 'man' as most similar
• Thinking output is an error or empty list

4. Identify the error in this Word2Vec training code and select the fix:

from gensim.models import Word2Vec
sentences = [['cat', 'dog', 'mouse'], ['dog', 'cat', 'elephant']]
model = Word2Vec(sentences, vector_size=5, window=2, min_count=2, epochs=5)
model.train(sentences, total_examples=1, epochs=5)

medium

A. Set min_count=1 to include all words in training

B. Change total_examples to len(sentences) in train()

C. Remove the train() call because Word2Vec trains on initialization

D. Increase vector_size to at least 10

Solution

1. Step 1: Analyze min_count effect

   min_count=2 excludes words appearing less than twice. 'mouse' and 'elephant' appear once, so they are ignored.

2. Step 2: Fix by lowering min_count

   Setting min_count=1 includes all words, fixing vocabulary size and training coverage.

3. Final Answer:

   Set min_count=1 to include all words in training -> Option A

4. Quick Check:

   min_count controls word inclusion [OK]

Hint: min_count filters words; lower it to include rare words [OK]

Common Mistakes:

• Ignoring min_count effect on vocabulary
• Thinking train() is unnecessary or wrongly used
• Changing vector_size without addressing vocabulary issue

5. You want to find the word that best fits the analogy: 'king' is to 'man' as 'queen' is to ____. Using Word2Vec, which operation would you perform to find this word?

hard

A. vector('king') - vector('man') + vector('queen')

B. vector('king') + vector('man') - vector('queen')

C. vector('queen') - vector('king') + vector('man')

D. vector('man') - vector('king') + vector('queen')

Solution

1. Step 1: Understand analogy vector math

   Word2Vec analogies use vector arithmetic: 'king' - 'man' + 'queen' approximates the vector for the missing word.

2. Step 2: Apply formula to find the answer

   Performing vector('king') - vector('man') + vector('queen') finds the word similar to 'queen' but related like 'king' is to 'man'.

3. Final Answer:

   vector('king') - vector('man') + vector('queen') -> Option A

4. Quick Check:

   Analogy = king - man + queen [OK]

Hint: Use king - man + queen to solve analogies [OK]

Common Mistakes:

• Mixing up vector addition and subtraction order
• Using unrelated vector combinations
• Ignoring analogy vector math concept