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

Summary

In this chapter, we have introduced how to export your trained model in TensorFlow and convert it into a usable format in TensorFlow.js. TensorFlow exports models mainly in two types of formats. One is SavedModel, which is a low-level format but it enables us to flexibly control variable mapping and share it with multiple graph definitions. If you want to construct a serialized model in a sufficiently optimized manner, SavedModel is the one you need to consider. Another format is Keras HDF5. As Keras is a pretty popular framework running on top of TensorFlow, exporting a model in Keras format could be the optimal solution for developers. Thanks to the limited number of configurations we need to set explicitly, that must be the easiest one to export the pretrained model into the file.

An exported file cannot be used directly in TensorFlow.js. There is a tool named tfjs...