org.tensorflow

Class SavedModelBundle

java.lang.Object

org.tensorflow.SavedModelBundle

All Implemented Interfaces:

AutoCloseable

public class SavedModelBundle
extends Object
implements AutoCloseable

SavedModelBundle represents a model loaded from storage.

The model consists of a description of the computation (a Graph), a Session with tensors (e.g., parameters or variables in the graph) initialized to values saved in storage, and a description of the model (a serialized representation of a MetaGraphDef protocol buffer).

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	SavedModelBundle.Loader Options for loading a SavedModel.

Method Summary

Modifier and Type	Method and Description
void	close() Releases resources (the Graph and Session) associated with the saved model bundle.
Graph	graph() Returns the graph that describes the computation performed by the model.
static SavedModelBundle	load(String exportDir, String... tags) Load a saved model from an export directory.
static SavedModelBundle.Loader	loader(String exportDir) Load a saved model.
byte[]	metaGraphDef() Returns the serialized MetaGraphDef protocol buffer associated with the saved model.
Session	session() Returns the Session with which to perform computation using the model.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

load

public static SavedModelBundle load(String exportDir,
                                    String... tags)

Load a saved model from an export directory. The model that is being loaded should be created using the Saved Model API.

This method is a shorthand for:

 SavedModelBundler.loader().withTags(tags).load();
 

Parameters:

exportDir - the directory path containing a saved model.

tags - the tags identifying the specific metagraphdef to load.

Returns:

a bundle containing the graph and associated session.

loader

public static SavedModelBundle.Loader loader(String exportDir)

Load a saved model.

Returns a Loader object that can set configuration options before actually loading the model,

Parameters:

exportDir - the directory path containing a saved model.

metaGraphDef

public byte[] metaGraphDef()

Returns the serialized MetaGraphDef protocol buffer associated with the saved model.

graph

public Graph graph()

Returns the graph that describes the computation performed by the model.

session

public Session session()

Returns the Session with which to perform computation using the model.

Returns:

the initialized session

close

public void close()

Releases resources (the Graph and Session) associated with the saved model bundle.

Specified by:

close in interface AutoCloseable