/MNIST-js

Training conv model for MNIST dataset on browser with tensorflowjs.

Primary LanguageJavaScript

MNIST-js

Goal

  • Training conv model for MNIST dataset on browser with tensorflowjs.

Result

  • Capture while training

To do

  • print status message
  • plot graph (loss, accucuracy) ? when batch end

How to build

  • cd into the directory

  • If you are using yarn:

      yarn && yarn watch

  • If you are using npm:

      npm install 
  npm run watch