Book Image

TensorFlow Machine Learning Projects

By : Ankit Jain, Amita Kapoor
Book Image

TensorFlow Machine Learning Projects

By: Ankit Jain, Amita Kapoor

Overview of this book

TensorFlow has transformed the way machine learning is perceived. TensorFlow Machine Learning Projects teaches you how to exploit the benefits—simplicity, efficiency, and flexibility—of using TensorFlow in various real-world projects. With the help of this book, you’ll not only learn how to build advanced projects using different datasets but also be able to tackle common challenges using a range of libraries from the TensorFlow ecosystem. To start with, you’ll get to grips with using TensorFlow for machine learning projects; you’ll explore a wide range of projects using TensorForest and TensorBoard for detecting exoplanets, TensorFlow.js for sentiment analysis, and TensorFlow Lite for digit classification. As you make your way through the book, you’ll build projects in various real-world domains, incorporating natural language processing (NLP), the Gaussian process, autoencoders, recommender systems, and Bayesian neural networks, along with trending areas such as Generative Adversarial Networks (GANs), capsule networks, and reinforcement learning. You’ll learn how to use the TensorFlow on Spark API and GPU-accelerated computing with TensorFlow to detect objects, followed by how to train and develop a recurrent neural network (RNN) model to generate book scripts. By the end of this book, you’ll have gained the required expertise to build full-fledged machine learning projects at work.
Table of Contents (23 chapters)
Title Page
Copyright and Credits
Dedication
About Packt
Contributors
Preface
Index

Summary


Machine learning is at the edge of the next wave, where we try to make ML ubiquitous in our everyday life. It has several advantages such as offline access, data privacy, and so on.

In this chapter, we looked at a new library from Google known as TensorFlow Lite, which has been optimized for deploying ML models on mobile and embedded devices. We understood the architecture of TensorFlow Lite, which converts the trained TensorFlow model into .tflite format. This is designed for inference at fast speed and low memory on devices. TensorFlow Lite also supports multiple platforms, such as Android, iOS, Linux, and Raspberry Pi.

Next, we used the MNIST handwritten digit dataset to train a deep learning model. Subsequently, we followed the necessary steps to convert the trained model into .tflite format. The steps are as follows:

  1. Froze the graph with variables converted to constants
  2. Optimized the graph for inference by removing the unused ops like Dropout
  3. Used TensorFlow Optimization Converter...