Model training component in TensorFlow - Model Metrics & Evaluation

Metrics & Evaluation - Model training component

Which metric matters for Model Training and WHY

When training a model, the key metrics to watch are loss and accuracy. Loss tells us how far off the model's predictions are from the true answers. Accuracy shows how many predictions are correct. We focus on these because they guide us to improve the model step-by-step during training.

Confusion Matrix Example

For classification tasks, a confusion matrix helps us see detailed results:

      | Predicted Positive | Predicted Negative |
      |--------------------|--------------------|
      | True Positive (TP)  | False Positive (FP) |
      | False Negative (FN) | True Negative (TN)  |
    

Example with 100 samples:

      TP = 40, FP = 10
      FN = 5,  TN = 45
      Total = 100
    

Precision vs Recall Tradeoff

During training, improving one metric can lower another. For example:

• Precision means when the model says "yes", it is usually right.
• Recall means the model finds most of the "yes" cases.

In training a spam filter, high precision is important to avoid marking good emails as spam. But in training a cancer detector, high recall is critical to catch all cancer cases, even if some false alarms happen.

Good vs Bad Metric Values in Model Training

Good: Loss decreases steadily, accuracy increases, and validation metrics improve without big gaps from training metrics.

Bad: Loss stays high or jumps around, accuracy is low or does not improve, or validation metrics are much worse than training (sign of overfitting).

Common Pitfalls in Model Training Metrics

• Accuracy Paradox: High accuracy can be misleading if data is unbalanced (e.g., 95% accuracy by always predicting the majority class).
• Data Leakage: If training data leaks into validation, metrics look unrealistically good.
• Overfitting: Training accuracy very high but validation accuracy low means the model memorizes training data but fails on new data.

Self Check

Your model has 98% accuracy but only 12% recall on fraud cases. Is it good for production?

Answer: No. The model misses most fraud cases (low recall), which is dangerous. Despite high accuracy, it fails the main goal of catching fraud. You need to improve recall before using it.

Key Result

Loss and accuracy guide model training progress; watch for balanced improvement and avoid overfitting.

Practice

1. What is the main purpose of the model.fit function in TensorFlow?

easy

A. To make predictions on new data

B. To evaluate the model's performance on test data

C. To save the model to disk

D. To train the model on the given data

Solution

1. Step 1: Understand the role of model.fit

   model.fit is used to train the model by adjusting its weights based on the training data.

2. Step 2: Differentiate from other functions

   Functions like model.evaluate check performance, model.save saves the model, and model.predict makes predictions. These are not for training.

3. Final Answer:

   To train the model on the given data -> Option D

4. Quick Check:

   model.fit = train model [OK]

Hint: Remember: fit means training the model [OK]

Common Mistakes:

• Confusing fit with predict
• Thinking fit saves the model
• Mixing up training and evaluation functions

2. Which of the following is the correct way to set the number of training cycles (epochs) in model.fit?

easy

A. model.fit(x_train, y_train, epoch=10)

B. model.fit(x_train, y_train, epochs=10)

C. model.fit(x_train, y_train, n_epochs=10)

D. model.fit(x_train, y_train, cycles=10)

Solution

1. Step 1: Recall the correct parameter name

   The parameter to set training cycles is epochs, plural.

2. Step 2: Check the options for correct syntax

   Only model.fit(x_train, y_train, epochs=10) uses epochs=10 correctly. Others use incorrect parameter names.

3. Final Answer:

   model.fit(x_train, y_train, epochs=10) -> Option B

4. Quick Check:

   Use epochs to set training cycles [OK]

Hint: Use 'epochs' plural to set training rounds [OK]

Common Mistakes:

• Using 'epoch' instead of 'epochs'
• Using 'n_epochs' or 'cycles' which are invalid
• Misspelling the parameter name

3. What will be the output of the following code snippet?

import tensorflow as tf
import numpy as np

x_train = np.array([[1], [2], [3], [4]])
y_train = np.array([2, 4, 6, 8])

model = tf.keras.Sequential([
  tf.keras.layers.Dense(1, input_shape=(1,))
])

model.compile(optimizer='sgd', loss='mse')
history = model.fit(x_train, y_train, epochs=2, verbose=0)

print(history.history['loss'])

medium

A. A list of two loss values decreasing over epochs

B. An error because of missing metrics

C. [0.5, 0.25]

D. Empty list []

Solution

1. Step 1: Understand the code training process

   The model is a simple linear layer trained with mean squared error loss for 2 epochs on a linear dataset.

2. Step 2: Predict the output of history.history['loss']

   This returns a list of loss values for each epoch, which should decrease as the model learns.

3. Final Answer:

   A list of two loss values decreasing over epochs -> Option A

4. Quick Check:

   Loss list decreases over epochs [OK]

Hint: Training loss list shows progress per epoch [OK]

Common Mistakes:

• Expecting exact loss numbers without running
• Thinking metrics are required for loss output
• Assuming loss list is empty or error

4. Identify the error in this TensorFlow training code:

model = tf.keras.Sequential([
  tf.keras.layers.Dense(1, input_shape=(1,))
])

model.compile(optimizer='adam', loss='mse')
model.fit(x_train, y_train, epoch=5)

medium

A. The parameter 'epoch' should be 'epochs' in model.fit

B. Missing optimizer in model.compile

C. Input shape should be (None, 1) instead of (1,)

D. Loss function 'mse' is invalid

Solution

1. Step 1: Check the parameters of model.fit

   The parameter to set training cycles is epochs, not epoch.

2. Step 2: Verify other parts of the code

   Optimizer and loss are correctly set. Input shape (1,) is valid for single feature input.

3. Final Answer:

   The parameter 'epoch' should be 'epochs' in model.fit -> Option A

4. Quick Check:

   Use 'epochs' plural in model.fit [OK]

Hint: Check spelling of 'epochs' parameter carefully [OK]

Common Mistakes:

• Using 'epoch' instead of 'epochs'
• Assuming input_shape is wrong for single feature
• Thinking 'mse' is invalid loss

5. You want to train a TensorFlow model on a dataset but notice the training loss does not decrease after many epochs. Which of the following is the best next step to improve training?

hard

A. Remove the loss function from model.compile

B. Increase the number of epochs to 1000 without changing anything else

C. Change the optimizer to a different one like 'adam' and check learning rate

D. Use model.predict instead of model.fit

Solution

1. Step 1: Understand why loss might not decrease

   Loss not decreasing often means the optimizer or learning rate is not suitable for the problem.

2. Step 2: Choose the best corrective action

   Changing to a better optimizer like 'adam' and tuning learning rate helps the model learn better. Increasing epochs alone won't help if learning is stuck.

3. Final Answer:

   Change the optimizer to a different one like 'adam' and check learning rate -> Option C

4. Quick Check:

   Adjust optimizer and learning rate to improve training [OK]

Hint: If loss stuck, try different optimizer and tune learning rate [OK]

Common Mistakes:

• Just increasing epochs without fixing optimizer
• Removing loss function causes errors
• Using predict instead of fit for training