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)