Counting 2,987 Big Data & Machine Learning Frameworks, Toolsets, and Examples...
Suggestion? Feedback? Tweet @stkim1

Last Commit
Jul. 5, 2018
Jul. 2, 2018


Have you ever woken up one day and thought: My life would be so much better if I could define and train my machine learning models in React. I haven't.

Define, train and visualize training of your ML models in your favorite front-end library (React) backed by your second favorite front-end library (Tensorflow.js).


  • Define models in React/JSX
  • Stream training data via ES6 Generators
  • Turn on training progress visualization with a single flag
  • Pause training

Demo App

Check out the live demo/clone the demo app to get started quicker.

out of the box training visualization

Out of the box training metric visualization!

Getting Started


yarn add react @tensorflow/tfjs tfjsx


npm install react @tensorflow/tfjs tfjsx

Write Some Machine Learning

import React from 'react';
import ReactDOM from 'react-dom';
import { Train, Model, Dense } from 'tfjsx';

// Define a generator of train data
function* trainDataGenerator() {
  yield { x: 1, y: 1 };
  yield { x: 4, y: 4 };
  yield { x: 8, y: 8 };

function MyTrainedModel() {
  // Train the model with the training data generator defined above
  return (
      onTrainEnd={model => model.describe()}
      {/* Define the model architecture */}
      <Model optimizer='sgd' loss='meanSquaredError'>
        <Dense units={1} inputShape={[1]} />

ReactDOM.render(<MyTrainedModel />, document.getElementById('app'));



Property Name Type Description
trainData function* () The generator should yield an object with x and y properties corresponding to training data and label.
validationData function* () Same as trainData, but should generate validation data. Will be used to output validation metrics during training.
epochs Number Number of epochs to train the model for
batchSize Number The number of samples to include in each training batch
samples Number Number of expected samples the generator will be able to generate.
onTrainEnd function(tf.Model) Called after the model is done training, the trained model is passed into the callback
onBatchEnd function(Object metrics, tf.Model) Called after each batch is done training, an object with that batch's training metrics along with the current model is passed into the callback.
train Bool Turn on and off training
display Bool Enable or disable graphing of training status


All valid config properties passed into model.compile are valid here. See config.


Similar to Model, all valid layers have their props passed through as properties of the config object in Tensorflow.js. The following layers are currently available:

Adding new layer types is simple, PRs are always welcome :)

Future Roadmap/Wishlist

  • Model summarization (adding display flag to Model)
  • Layer activation visualization (adding display flag to any layer)
  • Model evaluation visualizations
  • Allow pre-trained models to be used as a layer