Troy-MA / tfjs-firebase

Train 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 project covers:

  • Training model with Tensorflow.js inside Cloud Functions and with data from firestore (using only JavaScript).
  • Retraining the model if it was already trained and save to finish quickly and efficiently.
  • Making predictions with a cloud function.

I've created as simple as possible usecase. My model should predict if a comment (like YouTube one, or a Play store review) is positive or negative. I train it from Firestore database using NodeJS. You can fell free to clone the project and use it for any purpose. The code can be easly modified to predict star reviews, app statistics and much more.

Demos

Training

Training the model, gif

Predictions

Doing predictions with the model, gif

About

Train 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%