Module: tf.estimator.experimental

TensorFlow 2 version

Public API for tf.estimator.experimental namespace.

Classes

class InMemoryEvaluatorHook: Hook to run evaluation in training without a checkpoint.

class KMeans: An Estimator for K-Means clustering.

class LinearSDCA: Stochastic Dual Coordinate Ascent helper for linear estimators.

Functions

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

call_logit_fn(...): Calls logit_fn (experimental).

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

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

make_early_stopping_hook(...): Creates early-stopping hook.

make_stop_at_checkpoint_step_hook(...): Creates a proper StopAtCheckpointStepHook based on chief status.

stop_if_higher_hook(...): Creates hook to stop if the given metric is higher than the threshold.

stop_if_lower_hook(...): Creates hook to stop if the given metric is lower than the threshold.

stop_if_no_decrease_hook(...): Creates hook to stop if metric does not decrease within given max steps.

stop_if_no_increase_hook(...): Creates hook to stop if metric does not increase within given max steps.