Skip to content
master
Switch branches/tags
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
src
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Number Predictor using Tensorflow.js

Demo

A simple number predictor demo built using a Tensorflow.js model trained on the MNIST dataset.

Pre-trained Model

The pre-trained model used by default was trained locally on a Mac machine using Keras and then converted to a Tensorflow.js format.

Browser Trained Model

The local model is trained locally in-browser using Tensorflow.js.

Model Architecture

A Convolutional Neural Network (CNN) with 5 layers, 1 input 3 hidden and 1 for output. See code for more on activation function and other info on the model.

General Architecture

  1. Load the MNIST dataset if training the model in the browser.
  2. Load the pre-trained model or the browser trained model.
  3. Take the input from the canvas element and do some processing to the pixels.
  4. Take the input and run it throw the trained model.
  5. Take the predictions and determine which one has the higher probability.
  6. Display the highest predicted number.