Book Image

Python: Advanced Guide to Artificial Intelligence

By : Giuseppe Bonaccorso, Rajalingappaa Shanmugamani
Book Image

Python: Advanced Guide to Artificial Intelligence

By: Giuseppe Bonaccorso, Rajalingappaa Shanmugamani

Overview of this book

This Learning Path is your complete guide to quickly getting to grips with popular machine learning algorithms. You'll be introduced to the most widely used algorithms in supervised, unsupervised, and semi-supervised machine learning, and learn how to use them in the best possible manner. Ranging from Bayesian models to the MCMC algorithm to Hidden Markov models, this Learning Path will teach you how to extract features from your dataset and perform dimensionality reduction by making use of Python-based libraries. You'll bring the use of TensorFlow and Keras to build deep learning models, using concepts such as transfer learning, generative adversarial networks, and deep reinforcement learning. Next, you'll learn the advanced features of TensorFlow1.x, such as distributed TensorFlow with TF clusters, deploy production models with TensorFlow Serving. You'll implement different techniques related to object classification, object detection, image segmentation, and more. By the end of this Learning Path, you'll have obtained in-depth knowledge of TensorFlow, making you the go-to person for solving artificial intelligence problems This Learning Path includes content from the following Packt products: • Mastering Machine Learning Algorithms by Giuseppe Bonaccorso • Mastering TensorFlow 1.x by Armando Fandango • Deep Learning for Computer Vision by Rajalingappaa Shanmugamani
Table of Contents (31 chapters)
Title Page
About Packt
Contributors
Preface
19
Tensor Processing Units
Index

Saving and Restoring models in TensorFlow


You can save and restore the models and the variables in TensorFlow by one of the following two methods:

  • A saver object created from the tf.train.Saver class
  • A SavedModel format based object created from the tf.saved_model_builder.SavedModelBuilder class

Let us see both the methods in action.

Note

You can follow along with the code in the Jupyter notebook ch-11a_Saving_and_Restoring_TF_Models.

Saving and restoring all graph variables with the saver class

We proceed as follows:

  1. To use the saver class, first an object of this class is created:
saver = tf.train.Saver()
  1. The simplest way to save all the variables in a graph is to call the save() method with the following two parameters: the session object and the path to the file on the disk where the variables will be saved:
with tf.Session() as tfs:
   ...
   saver.save(tfs,"saved-models/model.ckpt")
  1.  To restore the variables, the restore() method is called:
with tf.Session() as tfs:
    saver.restore(tfs,"saved...