TensorFlow.js Examples
XOR vanilla neural network
Doodle Classifier
This repo is experimental and in progress. It is an "MNIST"-style classification example using the Google QuickDraw dataset, p5js, and tensorflow.js. It is loosely based on the tfjs MNIST example.
Reference
RoadMap
- Simplify model removing convolutional layers. The idea is for this to be a dropdead simple example that I can use to explain tensorflow.js and the layers API. I'd like to cover convolutional neural networks as a secondary example.
- Incorporate testing data. At the moment no any validation / testing data is included during training. For clarity of the example I might like to run the testing as a separate function. What do the results mean me if I give it
null
data? - Guess user drawings in real-time.
- Train with a much larger dataset.
- Save model using local storage or to JSON file.
- Bring the idea of a higher level
Classifier
class that wraps keras layers into ml5.