TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

Module: hub

TensorFlow Hub Library.


class ImageModuleInfo

class KerasLayer: Wraps a Hub module (or a similar callable) for TF2 as a Keras Layer.

class LatestModuleExporter: Regularly exports registered modules into timestamped directories.

class Module: Part of a TensorFlow model that can be transferred between models.

class ModuleSpec: Represents the contents of a Module before it has been instantiated.


add_signature(...): Adds a signature to the module definition.

attach_image_module_info(...): Attaches an ImageModuleInfo message from within a module_fn.

attach_message(...): Adds an attached message to the module definition.

create_module_spec(...): Creates a ModuleSpec from a function that builds the module's graph.

create_module_spec_from_saved_model(...): Experimental: Create a ModuleSpec out of a SavedModel.

get_expected_image_size(...): Returns expected [height, width] dimensions of an image input.

get_num_image_channels(...): Returns expected num_channels dimensions of an image input.

image_embedding_column(...): Uses a Module to get a dense 1-D representation from the pixels of images.

load(...): Loads a module from a handle.

load_module_spec(...): Loads a ModuleSpec from the filesystem.

register_module_for_export(...): Register a Module to be exported under export_name.

resolve(...): Resolves a module handle into a path.

sparse_text_embedding_column(...): Uses a Module to construct dense representations from sparse text features.

text_embedding_column(...): Uses a Module to construct a dense representation from a text feature.

Other Members