TensorFlow.js Example: MNIST CNN Transfer Learning Demo

This demo shows how to perform transfer learning using the Layers API of TensorFlow.js.

A simple convnet was trained in Python Keras on only the first 5 digits [0..4] from the MNIST dataset. The resulting model is hosted at a URL and loaded into TensorFlow.js using tf.loadLayersModel().
The convolutional layers are frozen, and the dense layers are fine-tuned in the browser to classify the digits [5..9].

To launch the demo, do

yarn
yarn watch