radi-cho / tfjs-firebase

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

License:Apache License 2.0


Languages

Language:JavaScript 100.0%