This demo shows how to perform classification on the classic Iris flower data set using the Layers API of TensorFlow.js.
It demonstrates ways to create a model:
- Loading a pretrained model hosted at a URL, using
tf.loadLayersModel() - Creating and training a model from scratch in the browser.
This demo also shows how to use the callbacks field of the Model.fit()
configuration to perform real-time visualization of training progress.
The model consists of two Dense layers: one with a relu activation followed
by another with a softmax activation.
To launch the demo, do
yarn
yarn watch