Skip to content
Starter kit for implementing a machine learning model in TensorFlow.js and growing the model garden
TypeScript Shell JavaScript
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
demo
scripts
src
.gitignore
README.md
package.json
rollup.config.js
run_tests.ts
tsconfig.json
tslint.json
yarn.lock

README.md

TensorFlow.js Model Starter

A starter pack for implementing mobile-first machine learning models.

Installation

# git clone https://github.com/tensorflow/tfjs-models
# cd tfjs-models

export MODEL=model
git clone https://github.com/sdll/tfjs-model-starter.git $MODEL
cd $MODEL
rm -r .git*

Checklist

  • Add a script to convert the model weights serialized in Python to the format supported by TensorFlow.js to scripts (see examples)
  • Customise rollup.config.js and package.json, replacing model-name with the name of your model and adding a short description
  • Give a poignant name to the main model class and adjust the tests accordingly
  • Make a beautiful demo of the main functionality (see the template) and showcase the results in the main and demo docs

Setup

Examples

You can’t perform that action at this time.