Book Image

Hands-On Machine Learning with TensorFlow.js

By : Kai Sasaki
Book Image

Hands-On Machine Learning with TensorFlow.js

By: Kai Sasaki

Overview of this book

TensorFlow.js is a framework that enables you to create performant machine learning (ML) applications that run smoothly in a web browser. With this book, you will learn how to use TensorFlow.js to implement various ML models through an example-based approach. Starting with the basics, you'll understand how ML models can be built on the web. Moving on, you will get to grips with the TensorFlow.js ecosystem to develop applications more efficiently. The book will then guide you through implementing ML techniques and algorithms such as regression, clustering, fast Fourier transform (FFT), and dimensionality reduction. You will later cover the Bellman equation to solve Markov decision process (MDP) problems and understand how it is related to reinforcement learning. Finally, you will explore techniques for deploying ML-based web applications and training models with TensorFlow Core. Throughout this ML book, you'll discover useful tips and tricks that will build on your knowledge. By the end of this book, you will be equipped with the skills you need to create your own web-based ML applications and fine-tune models to achieve high performance.
Table of Contents (17 chapters)
Free Chapter
1
Section 1: The Rationale of Machine Learning and the Usage of TensorFlow.js
5
Section 2: Real-World Applications of TensorFlow.js
12
Section 3: Productionizing Machine Learning Applications with TensorFlow.js

Converting models using tfjs-converter

Unfortunately, models such as SavedModel of HDF5 created by TensorFlow cannot be used in the world of TensorFlow.js directly. It is inevitable that you will have to convert the model into a format readable by the web platform.

Converting a TensorFlow SavedModel

Once the SavedModel is created, you can convert the SavedModel into TensorFlow.js format as follows:

$ tensorflowjs_converter \
--output_node_names=output \
--input_format=tf_saved_model \
./my_saved_model ./my_tfjs_model

The input path and output path are required as positional arguments. (my_saved_model and my_tfjs_model). The model will be generated in the my_tfjs_model directory. The options specified in the preceding...