Text embedding models in Prompt Engineering / GenAI - Model Pipeline Trace

Model Pipeline - Text embedding models

This pipeline turns words or sentences into numbers that computers can understand. These numbers capture the meaning of the text so machines can compare or use them in tasks like search or recommendations.

Data Flow - 5 Stages

1Raw Text Input

1000 sentences→Collect sentences or phrases as input→1000 sentences

"I love sunny days", "Machine learning is fun"

↓

2Text Cleaning

1000 sentences→Lowercase, remove punctuation, and extra spaces→1000 cleaned sentences

"i love sunny days", "machine learning is fun"

↓

3Tokenization

1000 cleaned sentences→Split sentences into words or tokens→1000 lists of tokens

["i", "love", "sunny", "days"], ["machine", "learning", "is", "fun"]

↓

4Embedding Lookup

1000 lists of tokens→Convert each token to a fixed-size vector using a trained embedding table→1000 lists of vectors (e.g., 1000 x variable length x 300)

[[0.1,0.3,...], [0.5,0.2,...], ...]

↓

5Pooling

1000 lists of vectors (variable length)→Combine token vectors into one vector per sentence (e.g., average)→1000 vectors (1000 x 300)

[0.3, 0.25, ..., 0.4]

Training Trace - Epoch by Epoch

Loss
0.9 |****
0.8 |*** 
0.7 |**  
0.6 |**  
0.5 |*   
0.4 |*   
0.3 |    
     1 2 3 4 5 Epochs

Epoch	Loss ↓	Accuracy ↑	Observation
1	0.85	0.40	Model starts learning basic word relationships.
2	0.65	0.55	Embeddings improve, capturing more meaning.
3	0.50	0.70	Model better understands word similarities.
4	0.40	0.78	Embeddings become more precise.
5	0.35	0.82	Training converges with good semantic capture.

Prediction Trace - 4 Layers

Layer 1: Input Sentence

Layer 2: Tokenization

Layer 3: Embedding Lookup

Layer 4: Pooling

Model Quiz - 3 Questions

What does the pooling step do in the embedding pipeline?

A Removes punctuation from text

B Splits sentences into words

C Combines word vectors into one sentence vector

D Converts words to lowercase

Key Insight

Text embedding models turn words into numbers that capture meaning. This helps machines understand and compare text easily. Training improves these numbers so similar words or sentences get similar vectors.

Practice

1. What is the main purpose of a text embedding model?

easy

A. To convert text into numbers that capture its meaning

B. To translate text from one language to another

C. To generate images from text descriptions

D. To count the number of words in a text

Solution

1. Step 1: Understand what text embedding models do

   Text embedding models turn words or sentences into number arrays that represent their meaning.

2. Step 2: Compare options with this understanding

   Only To convert text into numbers that capture its meaning describes converting text into meaningful numbers. Other options describe different tasks.

3. Final Answer:

   To convert text into numbers that capture its meaning -> Option A

4. Quick Check:

   Text embedding = convert text to meaningful numbers [OK]

Hint: Remember: embeddings turn text into numbers for meaning [OK]

Common Mistakes:

• Confusing embeddings with translation
• Thinking embeddings generate images
• Assuming embeddings just count words

2. Which of the following is the correct way to get an embedding vector from a text using a Python function get_embedding(text)?

easy

A. embedding = get_embedding->text

B. embedding = get_embedding[text]

C. embedding = get_embedding{text}

D. embedding = get_embedding(text)

Solution

1. Step 1: Recall Python function call syntax

   In Python, functions are called with parentheses and arguments inside, like func(arg).

2. Step 2: Match syntax with options

   Only embedding = get_embedding(text) uses parentheses correctly. Options A, B, and C use invalid syntax for function calls.

3. Final Answer:

   embedding = get_embedding(text) -> Option D

4. Quick Check:

   Function call uses parentheses () [OK]

Hint: Use parentheses () to call functions in Python [OK]

Common Mistakes:

• Using square brackets [] instead of parentheses
• Using curly braces {} instead of parentheses
• Using arrow -> instead of parentheses

3. Given the code below, what will be the output?

def dummy_embedding(text):
    return [len(text), sum(ord(c) for c in text) % 100]

result = dummy_embedding('cat')
print(result)

medium

A. [3, 12]

B. [3, 15]

C. [4, 30]

D. [3, 30]

Solution

1. Step 1: Calculate length of 'cat'

   The word 'cat' has 3 characters, so first element is 3.

2. Step 2: Calculate sum of ASCII codes modulo 100

   ord('c')=99, ord('a')=97, ord('t')=116; sum=99+97+116=312; 312 % 100 = 12.

3. Step 3: Determine output

   return [3, 12], so print([3, 12]).

4. Final Answer:

   [3, 12] -> Option A

5. Quick Check:

   len('cat')=3, (99+97+116)%100=12 [OK]

Hint: Calculate length and ASCII sum mod 100 carefully [OK]

Common Mistakes:

• Wrong ASCII sum calculation
• Miscounting string length
• Mixing uppercase and lowercase ASCII codes

4. The following code tries to get embeddings for two texts but doesn't work as intended. What is the problem?

def get_embedding(text):
    return [len(text)]

texts = ['hello', 'world']
embeddings = []
for t in texts:
    embeddings.append(get_embedding)
print(embeddings)

medium

A. The list texts is empty

B. The function is not called; it appends the function itself

C. The variable embeddings is not defined

D. The function get_embedding has wrong syntax

Solution

1. Step 1: Check the loop appending embeddings

   The code appends get_embedding without parentheses, so it adds the function object, not the result.

2. Step 2: Understand the problem

   Appending the function itself causes the list to hold function references, not embedding lists like [5] and [5].

3. Final Answer:

   The function is not called; it appends the function itself -> Option B

4. Quick Check:

   Missing () calls function, else appends function object [OK]

Hint: Add () to call function, not just reference it [OK]

Common Mistakes:

• Forgetting parentheses to call function
• Assuming list is empty causes error
• Thinking variable is undefined

5. You want to find the most similar sentence to 'I love apples' from a list using embeddings. Which approach is best?

hard

A. Count common words between 'I love apples' and each sentence

B. Translate all sentences to another language and compare lengths

C. Compute embeddings for all sentences, then find the one with smallest distance to 'I love apples' embedding

D. Randomly pick a sentence from the list

Solution

1. Step 1: Understand similarity with embeddings

   Embeddings turn sentences into number arrays capturing meaning, so comparing distances between embeddings finds similar sentences.

2. Step 2: Evaluate options for similarity search

   Compute embeddings for all sentences, then find the one with smallest distance to 'I love apples' embedding uses embeddings and distance, which is the correct method. Options A, C, and D do not use embeddings or meaningful similarity measures.

3. Final Answer:

   Compute embeddings for all sentences, then find the one with smallest distance to 'I love apples' embedding -> Option C

4. Quick Check:

   Use embeddings + distance for similarity [OK]

Hint: Use embedding distances to find similar texts [OK]

Common Mistakes:

• Using word count instead of embeddings
• Ignoring embeddings for similarity
• Random selection instead of comparison