tf.contrib.estimator

 

upper level

 

Modules

 

None

 

Classes

 

DNNEstimator:

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

DNNLinearCombinedEstimator:

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

LinearEstimator:

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

TowerOptimizer:

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

 

Functions

 

add_metrics(…):

Creates a new ${tf.estimator.Estimator} which has given metrics.

binary_classification_head(…):

Creates a _Head for single label binary classification.

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.

forward_features(…):

Forward features to predictions dictionary.

linear_logit_fn_builder(…):

Function builder for a linear logit_fn.

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.

regression_head(…):

Creates a _Head for regression using the mean_squared_error loss.

replicate_model_fn(…):

Replicate Estimator.model_fn over GPUs.