shiffman / tensorflow-js-examples

Working on some new examples with tensorflow.js and p5.js

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

• JS Doodle Classifier video tutorials
• ml4a ofx Doodle Classifier

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.

Neuro-Evolution

• Flappy Bird Demo: Learning
• Flappy Bird Demo: Loading Saved Model
• Steering Ecosystem Simulation Demo