Module: tf.estimator

Aliases:

• Module tf.compat.v1.estimator

Modules

experimental module

export module

inputs module

tpu module

Classes

class BaselineClassifier: A classifier that can establish a simple baseline.

class BaselineEstimator: An estimator that can establish a simple baseline.

class BaselineRegressor: A regressor that can establish a simple baseline.

class BestExporter: This class exports the serving graph and checkpoints of the best models.

class BoostedTreesClassifier: A Classifier for Tensorflow Boosted Trees models.

class BoostedTreesEstimator: An Estimator for Tensorflow Boosted Trees models.

class BoostedTreesRegressor: A Regressor for Tensorflow Boosted Trees models.

class CheckpointSaverHook: Saves checkpoints every N steps or seconds.

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

class DNNClassifier: A classifier for TensorFlow DNN models.

class DNNEstimator: An estimator for TensorFlow DNN models with user-specified head.

class DNNLinearCombinedClassifier: An estimator for TensorFlow Linear and DNN joined classification models.

class DNNLinearCombinedEstimator: An estimator for TensorFlow Linear and DNN joined models with custom head.

class DNNLinearCombinedRegressor: An estimator for TensorFlow Linear and DNN joined models for regression.

class DNNRegressor: A regressor for TensorFlow DNN models.

class Estimator: Estimator class to train and evaluate TensorFlow models.

class EstimatorSpec: Ops and objects returned from a model_fn and passed to an Estimator.

class EvalSpec: Configuration for the "eval" part for the train_and_evaluate call.

class Exporter: A class representing a type of model export.

class FeedFnHook: Runs feed_fn and sets the feed_dict accordingly.

class FinalExporter: This class exports the serving graph and checkpoints at the end.

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

class GlobalStepWaiterHook: Delays execution until global step reaches wait_until_step.

class LatestExporter: This class regularly exports the serving graph and checkpoints.

class LinearClassifier: Linear classifier model.

class LinearEstimator: An estimator for TensorFlow linear models with user-specified head.

class LinearRegressor: An estimator for TensorFlow Linear regression problems.

class LoggingTensorHook: Prints the given tensors every N local steps, every N seconds, or at end.

class ModeKeys: Standard names for Estimator model modes.

class NanLossDuringTrainingError

class NanTensorHook: Monitors the loss tensor and stops training if loss is NaN.

class ProfilerHook: Captures CPU/GPU profiling information every N steps or seconds.

class RunConfig: This class specifies the configurations for an Estimator run.

class SecondOrStepTimer: Timer that triggers at most once every N seconds or once every N steps.

class SessionRunArgs: Represents arguments to be added to a Session.run() call.

class SessionRunContext: Provides information about the session.run() call being made.

class SessionRunHook: Hook to extend calls to MonitoredSession.run().

class SessionRunValues: Contains the results of Session.run().

class StepCounterHook: Hook that counts steps per second.

class StopAtStepHook: Hook that requests stop at a specified step.

class SummarySaverHook: Saves summaries every N steps.

class TrainSpec: Configuration for the "train" part for the train_and_evaluate call.

class VocabInfo: Vocabulary information for warm-starting.

class WarmStartSettings: Settings for warm-starting in tf.estimator.Estimators.

Functions

add_metrics(...): Creates a new tf.estimator.Estimator which has given metrics.

classifier_parse_example_spec(...): Generates parsing spec for tf.parse_example to be used with classifiers.

regressor_parse_example_spec(...): Generates parsing spec for tf.parse_example to be used with regressors.

train_and_evaluate(...): Train and evaluate the estimator.