TensorFlow.js for browser - Model Metrics & Evaluation

Metrics & Evaluation - TensorFlow.js for browser

Which metric matters for TensorFlow.js in browser and WHY

When using TensorFlow.js in the browser, the key metrics depend on the task. For classification tasks, accuracy, precision, and recall are important to understand how well the model predicts classes. For regression, mean squared error (MSE) or mean absolute error (MAE) show prediction closeness.

Because browser models run on limited resources, inference speed and model size also matter. Fast predictions keep the app responsive, and small models load quickly for users.

Confusion matrix example for a browser classification model

    Actual \ Predicted | Positive | Negative
    -------------------|----------|---------
    Positive           |    40    |   10    
    Negative           |    5     |   45    
    

Here, True Positives (TP) = 40, False Negatives (FN) = 10, False Positives (FP) = 5, True Negatives (TN) = 45.

Precision = TP / (TP + FP) = 40 / (40 + 5) = 0.89

Recall = TP / (TP + FN) = 40 / (40 + 10) = 0.80

Accuracy = (TP + TN) / Total = (40 + 45) / 100 = 0.85

Precision vs Recall tradeoff in browser ML

Imagine a browser app that detects spam messages. If it marks too many good messages as spam (low precision), users get annoyed. If it misses many spam messages (low recall), spam floods inboxes.

High precision means fewer false alarms. High recall means catching most spam. Depending on user needs, you balance these.

TensorFlow.js models in browsers must balance this tradeoff while keeping fast response times.

Good vs Bad metric values for browser ML models

Good: Accuracy above 85%, precision and recall above 80%, and inference time under 100ms for smooth user experience.

Bad: Accuracy below 60%, precision or recall below 50%, or slow predictions causing UI lag.

Good metrics mean the model predicts well and runs fast enough for users.

Common pitfalls in TensorFlow.js browser model metrics

• Accuracy paradox: High accuracy can be misleading if classes are imbalanced.
• Data leakage: Using test data during training inflates metrics falsely.
• Overfitting: Model performs well on training but poorly on new data.
• Ignoring latency: Good accuracy but slow inference hurts user experience.

Self-check question

Your TensorFlow.js browser model has 98% accuracy but only 12% recall on fraud detection. Is it good for production?

Answer: No. The low recall means the model misses most fraud cases, which is dangerous. Despite high accuracy, it fails the main goal. You should improve recall before production.

Key Result

For TensorFlow.js browser models, balance accuracy, precision, recall, and inference speed to ensure good predictions and smooth user experience.

Practice

1. What is the main advantage of using TensorFlow.js in a web browser?

easy

A. Needs a separate app to visualize data

B. Requires installing special software on the user's computer

C. Run and train machine learning models directly in the browser without extra software

D. Only works on server-side environments

Solution

1. Step 1: Understand TensorFlow.js purpose

   TensorFlow.js is designed to run and train ML models directly in the browser environment.

2. Step 2: Compare options

   Options A, B, and C mention installing software, server-only use, or separate apps, which are not true for TensorFlow.js.

3. Final Answer:

   Run and train machine learning models directly in the browser without extra software -> Option C

4. Quick Check:

   TensorFlow.js runs ML in browser = D [OK]

Hint: TensorFlow.js runs ML models inside browsers without installs [OK]

Common Mistakes:

• Thinking TensorFlow.js needs server or software install
• Confusing browser use with server-side ML
• Assuming extra apps are required

2. Which of the following is the correct way to create a 1D tensor with values [1, 2, 3] in TensorFlow.js?

easy

A. tf.tensor([1; 2; 3])

B. tf.tensor1d(1, 2, 3)

C. tf.tensor([1 2 3])

D. tf.tensor([1, 2, 3])

Solution

1. Step 1: Recall TensorFlow.js tensor creation syntax

   The correct method to create a 1D tensor with values is tf.tensor([1, 2, 3]).

2. Step 2: Check other options for syntax errors

   tf.tensor1d(1, 2, 3) uses wrong function and arguments, A and D have invalid array syntax.

3. Final Answer:

   tf.tensor([1, 2, 3]) -> Option D

4. Quick Check:

   Correct tensor syntax = C [OK]

Hint: Use tf.tensor with array brackets and commas [OK]

Common Mistakes:

• Using semicolons or spaces instead of commas in arrays
• Wrong function name or arguments
• Missing array brackets

3. What will be the output of this TensorFlow.js code snippet?

const t = tf.tensor([1, 2, 3, 4]);
const result = t.square().arraySync();
console.log(result);

medium

A. [1, 4, 9, 16]

B. [1, 2, 3, 4]

C. [2, 4, 6, 8]

D. Error: square() is not a function

Solution

1. Step 1: Understand the code operations

   The tensor t contains [1, 2, 3, 4]. The square() method squares each element.

2. Step 2: Calculate squared values and output

   Squaring each element: 1²=1, 2²=4, 3²=9, 4²=16. The arraySync() converts tensor to JS array.

3. Final Answer:

   [1, 4, 9, 16] -> Option A

4. Quick Check:

   Tensor square() output = A [OK]

Hint: square() applies element-wise power of 2 [OK]

Common Mistakes:

• Confusing square() with identity (no change)
• Expecting sum instead of element-wise operation
• Thinking square() is not a valid method

4. Identify the error in this TensorFlow.js code snippet:

const t = tf.tensor([1, 2, 3]);
const reshaped = t.reshape([2, 2]);
console.log(reshaped.arraySync());

medium

A. tf.tensor() cannot create 1D tensors

B. The reshape size [2, 2] does not match the number of elements in tensor

C. arraySync() is not a valid method

D. reshape() requires a 1D array argument

Solution

1. Step 1: Check tensor size and reshape target shape

   The tensor has 3 elements, but reshape tries to make shape [2, 2] which needs 4 elements.

2. Step 2: Understand reshape requirements

   Reshape must keep total elements same. Here 3 != 4, so it causes an error.

3. Final Answer:

   The reshape size [2, 2] does not match the number of elements in tensor -> Option B

4. Quick Check:

   Reshape size must match elements = A [OK]

Hint: Reshape shape must multiply to original tensor size [OK]

Common Mistakes:

• Ignoring total element count mismatch
• Thinking arraySync() is invalid
• Believing reshape needs 1D array argument

5. You want to build a simple linear regression model in TensorFlow.js that predicts y from x. Which sequence of steps is correct to train the model in the browser?

hard

A. Define model layers, compile with optimizer and loss, prepare tensors for x and y, then call model.fit() with data

B. Directly call model.predict() on raw arrays without compiling or training

C. Use tf.tensor() to create model weights manually and skip model.fit() entirely

D. Train the model only on the server and send predictions to browser

Solution

1. Step 1: Understand model training workflow in TensorFlow.js

   First, define the model architecture with layers, then compile it specifying optimizer and loss function.

2. Step 2: Prepare input and output data as tensors and call model.fit()

   Training requires data tensors for inputs (x) and outputs (y), then call model.fit() to train the model in the browser.

3. Final Answer:

   Define model layers, compile with optimizer and loss, prepare tensors for x and y, then call model.fit() with data -> Option A

4. Quick Check:

   Model training steps = B [OK]

Hint: Define, compile, prepare tensors, then fit model [OK]

Common Mistakes:

• Skipping compile before training
• Trying to predict without training
• Manually setting weights without training
• Assuming training must happen on server