Missed TensorFlow World? Check out the recap. Learn more

Module: tf.contrib.estimator

View source on GitHub

estimator python module.

Importing from tensorflow.python.estimator is unsupported and will soon break!

Classes

class RNNClassifier: A classifier for TensorFlow RNN models.

class RNNEstimator: An Estimator for TensorFlow RNN models with user-specified head.

class SavedModelEstimator: Create an Estimator from a SavedModel.

class TowerOptimizer: Gathers gradients from all towers and reduces them in the last one.

Functions

DNNClassifierWithLayerAnnotations(...): A classifier for TensorFlow DNN models with layer annotations.

DNNRegressorWithLayerAnnotations(...): A regressor for TensorFlow DNN models with layer annotations.

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

binary_classification_head(...): Creates a _Head for single label binary classification.

boosted_trees_classifier_train_in_memory(...): Trains a boosted tree classifier with in memory dataset.

boosted_trees_regressor_train_in_memory(...): Trains a boosted tree regressor with in memory dataset.

build_raw_supervised_input_receiver_fn(...): Build a supervised_input_receiver_fn for raw features and labels.

build_supervised_input_receiver_fn_from_input_fn(...): Get a function that returns a SupervisedInputReceiver matching an input_fn.

call_logit_fn(...): Calls logit_fn.

clip_gradients_by_norm(...): Returns an optimizer which clips gradients before applying them.

dnn_logit_fn_builder(...): Function builder for a dnn logit_fn.

export_all_saved_models(...): Exports requested train/eval/predict graphs as separate SavedModels. (deprecated)

export_saved_model_for_mode(...): Exports a single train/eval/predict graph as a SavedModel. (deprecated)

forward_features(...): Forward features to predictions dictionary.

linear_logit_fn_builder(...): Function builder for a linear logit_fn.

logistic_regression_head(...): Creates a _Head for logistic regression.

multi_class_head(...): Creates a _Head for multi class classification.

multi_head(...): Creates a _Head for multi-objective learning.

multi_label_head(...): Creates a _Head for multi-label classification.

poisson_regression_head(...): Creates a _Head for poisson regression using tf.nn.log_poisson_loss.

regression_head(...): Creates a _Head for regression using the mean_squared_error loss.

replicate_model_fn(...): Replicate Estimator.model_fn over GPUs. (deprecated)