Data transformation (TF Transform) in TensorFlow - Model Pipeline Trace

Model Pipeline - Data transformation (TF Transform)

This pipeline shows how raw data is cleaned and changed using TensorFlow Transform before training a model. It makes data ready by scaling numbers and turning text into numbers.

Data Flow - 3 Stages

1Raw data input

1000 rows x 3 columns→Original data with features: age, income, city→1000 rows x 3 columns

{"age": 25, "income": 50000, "city": "New York"}

↓

2Apply TF Transform preprocessing

1000 rows x 3 columns→Scale 'age' and 'income'; convert 'city' to integer IDs→1000 rows x 3 columns

{"age_scaled": 0.25, "income_scaled": 0.5, "city_id": 2}

↓

3Split data for training and evaluation

1000 rows x 3 columns→Split 80% train, 20% eval→Train: 800 rows x 3 columns, Eval: 200 rows x 3 columns

Train sample: {"age_scaled": 0.25, "income_scaled": 0.5, "city_id": 2}

Training Trace - Epoch by Epoch

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

Epoch	Loss ↓	Accuracy ↑	Observation
1	0.65	0.60	Model starts learning with moderate loss and accuracy.
2	0.48	0.75	Loss decreases and accuracy improves as model learns.
3	0.35	0.82	Model continues to improve with better predictions.
4	0.28	0.87	Loss lowers further; accuracy nearing good performance.
5	0.22	0.90	Training converges with low loss and high accuracy.

Prediction Trace - 3 Layers

Layer 1: Input raw sample

Layer 2: TF Transform preprocessing

Layer 3: Model prediction

Model Quiz - 3 Questions

What does TF Transform do to the 'city' feature?

A Removes the city feature

B Scales city names between 0 and 1

C Converts city names to integer IDs

D Leaves city names as text

Key Insight

Using TF Transform to preprocess data ensures consistent and clean input for the model. Scaling numeric features and encoding categorical ones helps the model learn better and faster.

Practice

1. What is the main purpose of TF Transform in machine learning pipelines?

easy

A. To visualize the dataset statistics

B. To train the machine learning model faster

C. To apply consistent data transformations during both training and serving

D. To generate synthetic data samples

Solution

1. Step 1: Understand TF Transform's role

   TF Transform is designed to apply the same data transformations consistently during training and serving.

2. Step 2: Compare options

   Options A, B, and D describe other tasks unrelated to TF Transform's main purpose.

3. Final Answer:

   To apply consistent data transformations during both training and serving -> Option C

4. Quick Check:

   Consistent transformations = To apply consistent data transformations during both training and serving [OK]

Hint: TF Transform ensures same data changes in training and serving [OK]

Common Mistakes:

• Confusing TF Transform with model training
• Thinking TF Transform generates data
• Assuming it visualizes data statistics

2. Which of the following is the correct way to define a TF Transform preprocessing function that scales a numeric feature named age by its mean and variance?

easy

A. def preprocessing_fn(inputs):
age = tft.scale_to_z_score(inputs['age'])
return {'age_scaled': age}

B. def preprocessing_fn(inputs):
return {'age_scaled': inputs['age'] / 100}

C. def preprocessing_fn(inputs):
return {'age_scaled': inputs['age'] + 10}

D. def preprocessing_fn(inputs):
age = inputs['age'] * 2
return {'age_scaled': age}

Solution

1. Step 1: Identify correct TF Transform scaling function

   TF Transform provides tft.scale_to_z_score to scale features by mean and variance.

2. Step 2: Check code correctness

   def preprocessing_fn(inputs): age = tft.scale_to_z_score(inputs['age']) return {'age_scaled': age} uses tft.scale_to_z_score correctly; others use incorrect or arbitrary scaling.

3. Final Answer:

   def preprocessing_fn(inputs): age = tft.scale_to_z_score(inputs['age']) return {'age_scaled': age} -> Option A

4. Quick Check:

   Use tft.scale_to_z_score for mean-variance scaling [OK]

Hint: Use tft.scale_to_z_score for standard scaling [OK]

Common Mistakes:

• Using simple division instead of TF Transform scaling
• Adding arbitrary constants instead of scaling
• Multiplying without normalization

3. Given the following TF Transform preprocessing function, what will be the output for input {'income': 50000} if the dataset mean income is 40000 and stddev is 10000?

def preprocessing_fn(inputs):
    income_scaled = tft.scale_to_z_score(inputs['income'])
    return {'income_scaled': income_scaled}

medium

A. {'income_scaled': 0.0}

B. {'income_scaled': 0.5}

C. {'income_scaled': -1.0}

D. {'income_scaled': 1.0}

Solution

1. Step 1: Calculate z-score for income

   Z-score = (value - mean) / stddev = (50000 - 40000) / 10000 = 1.0

2. Step 2: Match output format

   The function returns {'income_scaled': z-score}, so output is {'income_scaled': 1.0}

3. Final Answer:

   {'income_scaled': 1.0} -> Option D

4. Quick Check:

   Z-score calculation = 1.0 [OK]

Hint: Z-score = (value - mean) / stddev for scaling [OK]

Common Mistakes:

• Forgetting to subtract mean
• Dividing by mean instead of stddev
• Mixing up positive and negative signs

4. You wrote this TF Transform preprocessing function but get an error during analysis:

def preprocessing_fn(inputs):
    age_bucketized = tft.bucketize(inputs['age'], num_buckets=5)
    return {'age_bucketized': age_bucketized}

What is the likely cause of the error?

medium

A. The input 'age' is not a numeric tensor

B. The function tft.bucketize requires 'boundaries' instead of 'num_buckets'

C. Missing import statement for tensorflow_transform as tft

D. The return value must be a list, not a dictionary

Solution

1. Step 1: Check tft.bucketize function signature

   tft.bucketize requires 'boundaries' argument, not 'num_buckets'.

2. Step 2: Identify error cause

   Using 'num_buckets' causes an error because it's not a valid parameter.

3. Final Answer:

   The function tft.bucketize requires 'boundaries' instead of 'num_buckets' -> Option B

4. Quick Check:

   tft.bucketize needs boundaries, not num_buckets [OK]

Hint: Use boundaries list for tft.bucketize, not num_buckets [OK]

Common Mistakes:

• Passing num_buckets instead of boundaries
• Forgetting to import tensorflow_transform
• Returning wrong data type

5. You want to apply TF Transform to prepare a dataset with numeric feature height and categorical feature color. You want to scale height and one-hot encode color. Which preprocessing function correctly applies these transformations?

hard

A. def preprocessing_fn(inputs):
height_scaled = tft.scale_to_z_score(inputs['height'])
color_vocab = tft.compute_and_apply_vocabulary(inputs['color'])
color_one_hot = tf.one_hot(color_vocab, depth=3)
return {'height_scaled': height_scaled, 'color_one_hot': color_one_hot}

B. def preprocessing_fn(inputs):
height_scaled = tft.scale_to_z_score(inputs['height'])
color_one_hot = tft.compute_and_apply_vocabulary(inputs['color'])
return {'height_scaled': height_scaled, 'color_one_hot': color_one_hot}

C. def preprocessing_fn(inputs):
height_scaled = inputs['height'] / 100
color_one_hot = tf.one_hot(inputs['color'], depth=3)
return {'height_scaled': height_scaled, 'color_one_hot': color_one_hot}

D. def preprocessing_fn(inputs):
height_scaled = tft.scale_to_z_score(inputs['height'])
color_one_hot = tf.one_hot(inputs['color'], depth=3)
return {'height_scaled': height_scaled, 'color_one_hot': color_one_hot}

Solution

1. Step 1: Scale numeric feature 'height'

   Use tft.scale_to_z_score to scale 'height' correctly.

2. Step 2: One-hot encode categorical 'color'

   First, map 'color' to integer indices using tft.compute_and_apply_vocabulary, then apply tf.one_hot with correct depth.

3. Step 3: Verify options

   def preprocessing_fn(inputs): height_scaled = tft.scale_to_z_score(inputs['height']) color_vocab = tft.compute_and_apply_vocabulary(inputs['color']) color_one_hot = tf.one_hot(color_vocab, depth=3) return {'height_scaled': height_scaled, 'color_one_hot': color_one_hot} correctly chains vocabulary computation and one-hot encoding; others miss steps or misuse functions.

4. Final Answer:

   def preprocessing_fn(inputs): height_scaled = tft.scale_to_z_score(inputs['height']) color_vocab = tft.compute_and_apply_vocabulary(inputs['color']) color_one_hot = tf.one_hot(color_vocab, depth=3) return {'height_scaled': height_scaled, 'color_one_hot': color_one_hot} -> Option A

5. Quick Check:

   Vocabulary then one-hot encoding = def preprocessing_fn(inputs): height_scaled = tft.scale_to_z_score(inputs['height']) color_vocab = tft.compute_and_apply_vocabulary(inputs['color']) color_one_hot = tf.one_hot(color_vocab, depth=3) return {'height_scaled': height_scaled, 'color_one_hot': color_one_hot} [OK]

Hint: Use vocabulary lookup before one-hot encoding categorical data [OK]

Common Mistakes:

• Applying one-hot directly on strings
• Skipping vocabulary computation
• Scaling numeric features incorrectly