Skip to content
Branch: master
Find file History
Fetching latest commit…
Cannot retrieve the latest commit at this time.

TensorFlow.js Example: Iris Classification

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 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 watch

See this example live!

You can’t perform that action at this time.