ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

Module: tf.compat.v1.train

Support for training models.

See the Training guide.


experimental module: Public API for tf.train.experimental namespace.

queue_runner module: Public API for tf.train.queue_runner namespace.


class AdadeltaOptimizer: Optimizer that implements the Adadelta algorithm.

class AdagradDAOptimizer: Adagrad Dual Averaging algorithm for sparse linear models.

class AdagradOptimizer: Optimizer that implements the Adagrad algorithm.

class AdamOptimizer: Optimizer that implements the Adam algorithm.

class BytesList: Container that holds repeated fundamental values of byte type in the tf.train.Feature message.

class Checkpoint: Groups trackable objects, saving and restoring them.

class CheckpointManager: Manages multiple checkpoints by keeping some and deleting unneeded ones.

class CheckpointOptions: Options for constructing a Checkpoint.

class CheckpointSaverHook: Saves checkpoints every N steps or seconds.

class CheckpointSaverListener: Interface for listeners that take action before or after checkpoint save.

class ChiefSessionCreator: Creates a tf.compat.v1.Session for a chief.

class ClusterDef: A ProtocolMessage

class ClusterSpec: Represents a cluster as a set of "tasks", organized into "jobs".

class Coordinator: A coordinator for threads.

class Example: An Example is a mostly-normalized data format for storing data for training and inference.

class ExponentialMovingAverage: Maintains moving averages of variables by employing an exponential decay.

class Feature: A Feature is a list which may hold zero or more values.

class FeatureList: Contains zero or more values of tf.train.Features.

class FeatureLists: Contains the mapping from name to tf.train.FeatureList.

class Features: Protocol message for describing the features of a tf.train.Example.

class FeedFnHook: Runs feed_fn and sets the feed_dict accordingly.

class FinalOpsHook: A hook which evaluates Tensors at the end of a session.

class FloatList: Container that holds repeated fundamental values of float type in the tf.train.Feature message.

class FtrlOptimizer: Optimizer that implements the FTRL algorithm.

class GlobalStepWaiterHook: Delays execution until global step reaches wait_until_step.

class GradientDescentOptimizer: Optimizer that implements the gradient descent algorithm.