MNIST_React_TensorFlowJS

This repo is a POC about using a trained Keras model with MNIST database with TensorFlow.js and React.js!

Downloads
10
Stars
65
Committers
2

mnist-react-tfjs

Number recognition with MNIST data (React.js and TensorFlow.js)

Predict numbers using MNIST database with TensorFlow.js and React.js!

author: Aral Roca [email protected]

DEMO

Requirements

  • react - Recommended ^16.8.6
  • react-dom - Recommended 16.8.6
  • @tensorflow/tfjs - Recommended ^1.1.0"

Getting started

npm install --save mnist-react-tfjs

Then:

import NumberRecognition from 'mnist-react-tfjs';

Code

With RenderProps:

<NumberRecognition
    width={400}
    height={400}
>
{
    (predictedNumber, predictions) => (
        <h1>{predictedNumber}</h1>
    )
}
</NumberRecognition>

Without RenderProps:

<NumberRecognition
    width={400}
    height={400}
    onPredict={
        (number, predictions) => 
        console.log(number, predictions)
    }
    onClear={() => console.log('onClear')}
/>