Skip to content

radi-cho/tfjs-firebase

Repository files navigation

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

About

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

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published