/tfjs-firebase

Train a model with data from Firestore, save it to Cloud Storage and make predictions in Cloud Functions - entirely using NodeJS

Primary LanguageJavaScriptApache License 2.0Apache-2.0

Bag of words text classification

This workshop covers:

  • Training a model with Tensorflow.js inside Cloud Functions using data from Firestore (JavaScript only).
  • Retraining a model; Saving and loading a model.
  • Making predictions with a cloud function.

The developed model can predict if a comment is positive or negative. We train it with data from the Firestore database. The code can be easily modified to predict star reviews, app statistics, etc.

Demo

Training

Training the model, gif

Predictions

Doing predictions with the model, gif