Book Image

Mastering TensorFlow 1.x

Book Image

Mastering TensorFlow 1.x

Overview of this book

TensorFlow is the most popular numerical computation library built from the ground up for distributed, cloud, and mobile environments. TensorFlow represents the data as tensors and the computation as graphs. This book is a comprehensive guide that lets you explore the advanced features of TensorFlow 1.x. Gain insight into TensorFlow Core, Keras, TF Estimators, TFLearn, TF Slim, Pretty Tensor, and Sonnet. Leverage the power of TensorFlow and Keras to build deep learning models, using concepts such as transfer learning, generative adversarial networks, and deep reinforcement learning. Throughout the book, you will obtain hands-on experience with varied datasets, such as MNIST, CIFAR-10, PTB, text8, and COCO-Images. You will learn the advanced features of TensorFlow1.x, such as distributed TensorFlow with TF Clusters, deploy production models with TensorFlow Serving, and build and deploy TensorFlow models for mobile and embedded devices on Android and iOS platforms. You will see how to call TensorFlow and Keras API within the R statistical software, and learn the required techniques for debugging when the TensorFlow API-based code does not work as expected. The book helps you obtain in-depth knowledge of TensorFlow, making you the go-to person for solving artificial intelligence problems. By the end of this guide, you will have mastered the offerings of TensorFlow and Keras, and gained the skills you need to build smarter, faster, and efficient machine learning and deep learning systems.
Table of Contents (21 chapters)
19
Tensor Processing Units

Saving and restoring Keras models

In Keras, saving and restoring models is very simple. Keras provides three options:

  • Save the complete model with its network architecture, weights (parameters), training configuration, and optimizer state.
  • Save only the architecture.
  • Save only the weights.

For saving the complete model, use the model.save(filepath) function. This will save the complete model in an HDF5 file. The saved model can be loaded back using the keras.models.load_model(filepath) function. This function loads everything back, and then also compiles the model.

For saving the architecture of a model, use either the model.to_json() or model.to_yaml() function. These functions return a string that can be written to the disk file. While restoring the architecture, the string can be read back and the model architecture restored using the keras.models.model_from_json(json_string...