TensorFlow.js Example: Sequence-to-Sequence English-French Translation

Overview

This is the code for this video on Youtube by Siraj Raval on Translation systmes using NodeJS and TensorflowJS. The notes are in the jupyter notebook.

This demo shows how to perform sequence-to-sequence prediction using the Layers API of TensorFlow.js.

It demonstrates loading a pretrained model hosted at a URL, using tf.loadModel()

The model was trained in Python Keras, based on the lstm_seq2seq example. The training data was 149,861 English-French sentence pairs available from http://www.manythings.org/anki.

To launch the demo, do

yarn
yarn watch

See this example live!

Credits

credits for the code go to google, i've merely created a wrapper to get people started.