Full Code of tensorflow/estimator for AI

gitextract_a04upojl/

├── .bazelrc
├── .gitignore
├── BUILD
├── CONTRIBUTING.md
├── LICENSE
├── README.md
├── WORKSPACE
├── tensorflow_estimator/
│   ├── BUILD
│   ├── estimator.bzl
│   ├── python/
│   │   └── estimator/
│   │       ├── BUILD
│   │       ├── api/
│   │       │   ├── BUILD
│   │       │   ├── api_gen.bzl
│   │       │   ├── extractor_wrapper.py
│   │       │   └── generator_wrapper.py
│   │       ├── canned/
│   │       │   ├── __init__.py
│   │       │   ├── baseline.py
│   │       │   ├── baseline_estimator_test.py
│   │       │   ├── baseline_test.py
│   │       │   ├── canned_estimator_ds_integration_test.py
│   │       │   ├── dnn.py
│   │       │   ├── dnn_estimator_test.py
│   │       │   ├── dnn_linear_combined.py
│   │       │   ├── dnn_linear_combined_estimator_test.py
│   │       │   ├── dnn_linear_combined_test.py
│   │       │   ├── dnn_test_fc_v2.py
│   │       │   ├── dnn_testing_utils.py
│   │       │   ├── head.py
│   │       │   ├── head_test.py
│   │       │   ├── kmeans.py
│   │       │   ├── kmeans_test.py
│   │       │   ├── linear.py
│   │       │   ├── linear_estimator_test.py
│   │       │   ├── linear_model_test.py
│   │       │   ├── linear_optimizer/
│   │       │   │   ├── BUILD
│   │       │   │   ├── __init__.py
│   │       │   │   ├── doc/
│   │       │   │   │   └── sdca.ipynb
│   │       │   │   └── python/
│   │       │   │       ├── sdca_test.py
│   │       │   │       └── utils/
│   │       │   │           ├── sdca_ops.py
│   │       │   │           ├── sdca_ops_test.py
│   │       │   │           ├── sharded_mutable_dense_hashtable.py
│   │       │   │           └── sharded_mutable_dense_hashtable_test.py
│   │       │   ├── linear_test.py
│   │       │   ├── linear_testing_utils.py
│   │       │   ├── metric_keys.py
│   │       │   ├── optimizers.py
│   │       │   ├── optimizers_test.py
│   │       │   ├── optimizers_test_v2.py
│   │       │   ├── parsing_utils.py
│   │       │   ├── parsing_utils_test.py
│   │       │   ├── prediction_keys.py
│   │       │   ├── rnn.py
│   │       │   ├── rnn_test.py
│   │       │   ├── saved_model_estimator.py
│   │       │   ├── saved_model_estimator_test.py
│   │       │   ├── testdata/
│   │       │   │   └── wire_vocabulary.txt
│   │       │   ├── timeseries/
│   │       │   │   ├── BUILD
│   │       │   │   ├── ar_model.py
│   │       │   │   ├── ar_model_test.py
│   │       │   │   ├── ar_model_training_test.py
│   │       │   │   ├── estimators.py
│   │       │   │   ├── estimators_test.py
│   │       │   │   ├── feature_keys.py
│   │       │   │   ├── head.py
│   │       │   │   ├── head_test.py
│   │       │   │   ├── math_utils.py
│   │       │   │   ├── math_utils_test.py
│   │       │   │   ├── model.py
│   │       │   │   ├── model_utils.py
│   │       │   │   ├── saved_model_utils.py
│   │       │   │   └── state_management.py
│   │       │   └── v1/
│   │       │       ├── __init__.py
│   │       │       ├── baseline_estimator_test_v1.py
│   │       │       ├── baseline_test_v1.py
│   │       │       ├── dnn_estimator_test_v1.py
│   │       │       ├── dnn_linear_combined_estimator_test_v1.py
│   │       │       ├── dnn_linear_combined_test_v1.py
│   │       │       ├── dnn_test_fc_v1_v1.py
│   │       │       ├── dnn_test_fc_v2_v1.py
│   │       │       ├── dnn_testing_utils_v1.py
│   │       │       ├── linear_estimator_test_v1.py
│   │       │       ├── linear_test_v1.py
│   │       │       └── linear_testing_utils_v1.py
│   │       ├── distribute_strategy_estimator_integration_test.py
│   │       ├── distribute_strategy_estimator_training_test.py
│   │       ├── early_stopping.py
│   │       ├── early_stopping_test.py
│   │       ├── estimator.py
│   │       ├── estimator_export.py
│   │       ├── estimator_export_test.py
│   │       ├── estimator_lib.py
│   │       ├── estimator_test.py
│   │       ├── export/
│   │       │   ├── __init__.py
│   │       │   ├── export.py
│   │       │   ├── export_lib.py
│   │       │   ├── export_output.py
│   │       │   ├── export_test.py
│   │       │   ├── function.py
│   │       │   └── function_test.py
│   │       ├── exporter.py
│   │       ├── exporter_test.py
│   │       ├── extenders.py
│   │       ├── extenders_test.py
│   │       ├── gc.py
│   │       ├── gc_test.py
│   │       ├── head/
│   │       │   ├── __init__.py
│   │       │   ├── base_head.py
│   │       │   ├── base_head_test.py
│   │       │   ├── binary_class_head.py
│   │       │   ├── binary_class_head_test.py
│   │       │   ├── head_utils.py
│   │       │   ├── multi_class_head.py
│   │       │   ├── multi_class_head_test.py
│   │       │   ├── multi_head.py
│   │       │   ├── multi_head_test.py
│   │       │   ├── multi_label_head.py
│   │       │   ├── multi_label_head_test.py
│   │       │   ├── regression_head.py
│   │       │   ├── regression_head_test.py
│   │       │   ├── sequential_head.py
│   │       │   └── sequential_head_test.py
│   │       ├── hooks/
│   │       │   ├── __init__.py
│   │       │   ├── basic_session_run_hooks.py
│   │       │   ├── basic_session_run_hooks_test.py
│   │       │   ├── fake_summary_writer.py
│   │       │   ├── hooks.py
│   │       │   ├── hooks_test.py
│   │       │   └── session_run_hook.py
│   │       ├── inputs/
│   │       │   ├── __init__.py
│   │       │   ├── inputs.py
│   │       │   ├── numpy_io.py
│   │       │   ├── numpy_io_test.py
│   │       │   ├── pandas_io.py
│   │       │   ├── pandas_io_test.py
│   │       │   └── queues/
│   │       │       ├── __init__.py
│   │       │       ├── feeding_functions.py
│   │       │       ├── feeding_functions_test.py
│   │       │       ├── feeding_queue_runner.py
│   │       │       └── feeding_queue_runner_test.py
│   │       ├── keras_distribute_strategy_test.py
│   │       ├── keras_lib.py
│   │       ├── keras_premade_model_test.py
│   │       ├── keras_test.py
│   │       ├── mode_keys.py
│   │       ├── model_fn.py
│   │       ├── model_fn_test.py
│   │       ├── object_checkpointing_test.py
│   │       ├── run_config.py
│   │       ├── run_config_test.py
│   │       ├── tf_estimator_doctest.py
│   │       ├── tools/
│   │       │   ├── __init__.py
│   │       │   ├── analytics.py
│   │       │   ├── checkpoint_converter.py
│   │       │   └── checkpoint_converter_test.py
│   │       ├── tpu/
│   │       │   ├── BUILD
│   │       │   ├── __init__.py
│   │       │   ├── _tpu_estimator_embedding.py
│   │       │   ├── autotuning_iterations_per_loop_test.py
│   │       │   ├── error_handling.py
│   │       │   ├── error_handling_test.py
│   │       │   ├── iteration_count_estimator.py
│   │       │   ├── spatial_partitioning_api.md
│   │       │   ├── tpu_config.py
│   │       │   ├── tpu_config_test.py
│   │       │   ├── tpu_context.py
│   │       │   ├── tpu_enqueue_sequence_test.py
│   │       │   ├── tpu_estimator.py
│   │       │   ├── tpu_estimator_embedding_test.py
│   │       │   ├── tpu_estimator_evaluation_test.py
│   │       │   ├── tpu_estimator_export_test.py
│   │       │   ├── tpu_estimator_gradients_test.py
│   │       │   ├── tpu_estimator_input_v2_test.py
│   │       │   ├── tpu_estimator_integration_test.py
│   │       │   ├── tpu_estimator_model_parallelism_test.py
│   │       │   ├── tpu_estimator_signals_test.py
│   │       │   ├── tpu_estimator_test.py
│   │       │   └── util.py
│   │       ├── training.py
│   │       ├── training_test.py
│   │       ├── util.py
│   │       └── util_test.py
│   └── tools/
│       └── pip_package/
│           ├── BUILD
│           ├── build_pip_package.sh
│           ├── create_pip_helper.py
│           └── setup.py
└── third_party/
    └── py/
        ├── BUILD
        ├── BUILD.tpl
        └── python_configure.bzl

SYMBOL INDEX (4635 symbols across 144 files)

FILE: tensorflow_estimator/python/estimator/canned/baseline.py
  function _get_weight_column_key (line 70) | def _get_weight_column_key(weight_column):
  function _get_weight_column_key_v2 (line 81) | def _get_weight_column_key_v2(weight_column):
  function _get_batch_size_and_size_checks (line 92) | def _get_batch_size_and_size_checks(features, weight_column_key):
  function _baseline_logit_fn_builder (line 114) | def _baseline_logit_fn_builder(num_outputs, weight_column=None):
  function _baseline_model_fn (line 156) | def _baseline_model_fn(features,
  function _baseline_model_fn_builder_v2 (line 204) | def _baseline_model_fn_builder_v2(features, num_outputs, weight_column=N...
  function _baseline_model_fn_v2 (line 229) | def _baseline_model_fn_v2(
  class BaselineClassifierV2 (line 293) | class BaselineClassifierV2(estimator.EstimatorV2):
    method __init__ (line 346) | def __init__(self,
  class BaselineClassifier (line 404) | class BaselineClassifier(estimator.Estimator):
    method __init__ (line 407) | def __init__(self,
  class BaselineEstimatorV2 (line 433) | class BaselineEstimatorV2(estimator.EstimatorV2):
    method __init__ (line 483) | def __init__(self, head, model_dir=None, optimizer='Ftrl', config=None):
  class BaselineEstimator (line 512) | class BaselineEstimator(estimator.Estimator):
    method __init__ (line 515) | def __init__(self, head, model_dir=None, optimizer='Ftrl', config=None):
  class BaselineRegressorV2 (line 531) | class BaselineRegressorV2(estimator.EstimatorV2):
    method __init__ (line 579) | def __init__(self,
  class BaselineRegressor (line 627) | class BaselineRegressor(estimator.Estimator):
    method __init__ (line 630) | def __init__(self,

FILE: tensorflow_estimator/python/estimator/canned/baseline_estimator_test.py
  function assert_close (line 40) | def assert_close(expected, actual, rtol=1e-04, name='assert_close'):
  function save_variables_to_ckpt (line 55) | def save_variables_to_ckpt(model_dir):
  function _baseline_estimator_fn (line 62) | def _baseline_estimator_fn(weight_column=None, label_dimension=1, **kwar...
  function mock_optimizer_v2 (line 71) | def mock_optimizer_v2(testcase, expected_loss=None):
  class BaselineEstimatorEvaluationTest (line 114) | class BaselineEstimatorEvaluationTest(tf.test.TestCase):
    method setUp (line 116) | def setUp(self):
    method tearDown (line 119) | def tearDown(self):
    method test_evaluation_batch (line 124) | def test_evaluation_batch(self):
    method test_evaluation_weights (line 149) | def test_evaluation_weights(self):
    method test_evaluation_for_multi_dimensions (line 179) | def test_evaluation_for_multi_dimensions(self):
  class BaselineEstimatorPredictTest (line 208) | class BaselineEstimatorPredictTest(tf.test.TestCase):
    method setUp (line 210) | def setUp(self):
    method tearDown (line 213) | def tearDown(self):
    method test_1d (line 218) | def test_1d(self):
    method testMultiDim (line 238) | def testMultiDim(self):
  class BaselineEstimatorIntegrationTest (line 264) | class BaselineEstimatorIntegrationTest(tf.test.TestCase):
    method setUp (line 266) | def setUp(self):
    method tearDown (line 269) | def tearDown(self):
    method _test_complete_flow (line 274) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method test_numpy_input_fn (line 304) | def test_numpy_input_fn(self):
  class BaselineEstimatorTrainingTest (line 341) | class BaselineEstimatorTrainingTest(tf.test.TestCase):
    method setUp (line 343) | def setUp(self):
    method tearDown (line 346) | def tearDown(self):
    method _assert_checkpoint (line 351) | def _assert_checkpoint(self,
    method testFromScratch (line 371) | def testFromScratch(self):
    method testFromCheckpoint (line 390) | def testFromCheckpoint(self):

FILE: tensorflow_estimator/python/estimator/canned/baseline_test.py
  function assert_close (line 56) | def assert_close(expected, actual, rtol=1e-04, name='assert_close'):
  function save_variables_to_ckpt (line 71) | def save_variables_to_ckpt(model_dir):
  function queue_parsed_features (line 78) | def queue_parsed_features(feature_map):
  function sorted_key_dict (line 93) | def sorted_key_dict(unsorted_dict):
  function sigmoid (line 97) | def sigmoid(x):
  function _baseline_regressor_fn (line 101) | def _baseline_regressor_fn(*args, **kwargs):
  function _baseline_classifier_fn (line 105) | def _baseline_classifier_fn(*args, **kwargs):
  function mock_optimizer_v2 (line 109) | def mock_optimizer_v2(testcase, expected_loss=None):
  class BaselineRegressorEvaluationTest (line 157) | class BaselineRegressorEvaluationTest(tf.test.TestCase):
    method setUp (line 159) | def setUp(self):
    method tearDown (line 162) | def tearDown(self):
    method test_evaluation_for_simple_data (line 167) | def test_evaluation_for_simple_data(self):
    method test_evaluation_batch (line 190) | def test_evaluation_batch(self):
    method test_evaluation_weights (line 217) | def test_evaluation_weights(self):
    method test_evaluation_for_multi_dimensions (line 247) | def test_evaluation_for_multi_dimensions(self):
  class BaselineRegressorPredictTest (line 276) | class BaselineRegressorPredictTest(tf.test.TestCase):
    method setUp (line 278) | def setUp(self):
    method tearDown (line 281) | def tearDown(self):
    method test_1d (line 286) | def test_1d(self):
    method testMultiDim (line 306) | def testMultiDim(self):
  class BaselineRegressorIntegrationTest (line 332) | class BaselineRegressorIntegrationTest(tf.test.TestCase):
    method setUp (line 334) | def setUp(self):
    method tearDown (line 337) | def tearDown(self):
    method _test_complete_flow (line 342) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method test_numpy_input_fn (line 372) | def test_numpy_input_fn(self):
    method test_pandas_input_fn (line 408) | def test_pandas_input_fn(self):
    method test_input_fn_from_parse_example (line 437) | def test_input_fn_from_parse_example(self):
  class BaselineRegressorTrainingTest (line 498) | class BaselineRegressorTrainingTest(tf.test.TestCase):
    method setUp (line 500) | def setUp(self):
    method tearDown (line 503) | def tearDown(self):
    method _assert_checkpoint (line 508) | def _assert_checkpoint(self,
    method testFromScratchWithDefaultOptimizer (line 528) | def testFromScratchWithDefaultOptimizer(self):
    method testTrainWithOneDimLabel (line 542) | def testTrainWithOneDimLabel(self):
    method testTrainWithOneDimWeight (line 559) | def testTrainWithOneDimWeight(self):
    method testFromScratch (line 582) | def testFromScratch(self):
    method testFromCheckpoint (line 603) | def testFromCheckpoint(self):
    method testFromCheckpointMultiBatch (line 635) | def testFromCheckpointMultiBatch(self):
  class BaselineClassifierTrainingTest (line 675) | class BaselineClassifierTrainingTest(tf.test.TestCase):
    method setUp (line 677) | def setUp(self):
    method tearDown (line 680) | def tearDown(self):
    method _assert_checkpoint (line 684) | def _assert_checkpoint(self,
    method _testFromScratchWithDefaultOptimizer (line 706) | def _testFromScratchWithDefaultOptimizer(self, n_classes):
    method testBinaryClassesFromScratchWithDefaultOptimizer (line 720) | def testBinaryClassesFromScratchWithDefaultOptimizer(self):
    method testMultiClassesFromScratchWithDefaultOptimizer (line 723) | def testMultiClassesFromScratchWithDefaultOptimizer(self):
    method _testTrainWithTwoDimsLabel (line 726) | def _testTrainWithTwoDimsLabel(self, n_classes):
    method testBinaryClassesTrainWithTwoDimsLabel (line 745) | def testBinaryClassesTrainWithTwoDimsLabel(self):
    method testMultiClassesTrainWithTwoDimsLabel (line 748) | def testMultiClassesTrainWithTwoDimsLabel(self):
    method _testTrainWithOneDimLabel (line 751) | def _testTrainWithOneDimLabel(self, n_classes):
    method testBinaryClassesTrainWithOneDimLabel (line 768) | def testBinaryClassesTrainWithOneDimLabel(self):
    method testMultiClassesTrainWithOneDimLabel (line 771) | def testMultiClassesTrainWithOneDimLabel(self):
    method _testTrainWithTwoDimsWeight (line 774) | def _testTrainWithTwoDimsWeight(self, n_classes):
    method testBinaryClassesTrainWithTwoDimsWeight (line 796) | def testBinaryClassesTrainWithTwoDimsWeight(self):
    method testMultiClassesTrainWithTwoDimsWeight (line 799) | def testMultiClassesTrainWithTwoDimsWeight(self):
    method _testTrainWithOneDimWeight (line 802) | def _testTrainWithOneDimWeight(self, n_classes):
    method testBinaryClassesTrainWithOneDimWeight (line 822) | def testBinaryClassesTrainWithOneDimWeight(self):
    method testMultiClassesTrainWithOneDimWeight (line 825) | def testMultiClassesTrainWithOneDimWeight(self):
    method _testFromScratch (line 828) | def _testFromScratch(self, n_classes):
    method testBinaryClassesFromScratch (line 862) | def testBinaryClassesFromScratch(self):
    method testMultiClassesFromScratch (line 865) | def testMultiClassesFromScratch(self):
    method _testFromCheckpoint (line 868) | def _testFromCheckpoint(self, n_classes):
    method testBinaryClassesFromCheckpoint (line 918) | def testBinaryClassesFromCheckpoint(self):
    method testMultiClassesFromCheckpoint (line 921) | def testMultiClassesFromCheckpoint(self):
    method _testFromCheckpointFloatLabels (line 924) | def _testFromCheckpointFloatLabels(self, n_classes):
    method testBinaryClassesFromCheckpointFloatLabels (line 960) | def testBinaryClassesFromCheckpointFloatLabels(self):
    method testMultiClassesFromCheckpointFloatLabels (line 963) | def testMultiClassesFromCheckpointFloatLabels(self):
    method _testFromCheckpointMultiBatch (line 966) | def _testFromCheckpointMultiBatch(self, n_classes):
    method testBinaryClassesFromCheckpointMultiBatch (line 1024) | def testBinaryClassesFromCheckpointMultiBatch(self):
    method testMultiClassesFromCheckpointMultiBatch (line 1027) | def testMultiClassesFromCheckpointMultiBatch(self):
  class BaselineClassifierEvaluationTest (line 1031) | class BaselineClassifierEvaluationTest(tf.test.TestCase):
    method setUp (line 1033) | def setUp(self):
    method tearDown (line 1036) | def tearDown(self):
    method _test_evaluation_for_simple_data (line 1040) | def _test_evaluation_for_simple_data(self, n_classes):
    method test_binary_classes_evaluation_for_simple_data (line 1094) | def test_binary_classes_evaluation_for_simple_data(self):
    method test_multi_classes_evaluation_for_simple_data (line 1097) | def test_multi_classes_evaluation_for_simple_data(self):
    method _test_evaluation_batch (line 1100) | def _test_evaluation_batch(self, n_classes):
    method test_binary_classes_evaluation_batch (line 1164) | def test_binary_classes_evaluation_batch(self):
    method test_multi_classes_evaluation_batch (line 1167) | def test_multi_classes_evaluation_batch(self):
    method _test_evaluation_weights (line 1170) | def _test_evaluation_weights(self, n_classes):
    method test_binary_classes_evaluation_weights (line 1249) | def test_binary_classes_evaluation_weights(self):
    method test_multi_classes_evaluation_weights (line 1252) | def test_multi_classes_evaluation_weights(self):
  class BaselineClassifierPredictTest (line 1256) | class BaselineClassifierPredictTest(tf.test.TestCase):
    method setUp (line 1258) | def setUp(self):
    method tearDown (line 1261) | def tearDown(self):
    method _testPredictions (line 1265) | def _testPredictions(self, n_classes, label_vocabulary, label_output_fn):
    method testBinaryClassesWithoutLabelVocabulary (line 1332) | def testBinaryClassesWithoutLabelVocabulary(self):
    method testBinaryClassesWithLabelVocabulary (line 1339) | def testBinaryClassesWithLabelVocabulary(self):
    method testMultiClassesWithoutLabelVocabulary (line 1346) | def testMultiClassesWithoutLabelVocabulary(self):
    method testMultiClassesWithLabelVocabulary (line 1353) | def testMultiClassesWithLabelVocabulary(self):
  class BaselineClassifierIntegrationTest (line 1361) | class BaselineClassifierIntegrationTest(tf.test.TestCase):
    method setUp (line 1363) | def setUp(self):
    method tearDown (line 1366) | def tearDown(self):
    method _test_complete_flow (line 1370) | def _test_complete_flow(self, n_classes, train_input_fn, eval_input_fn,
    method _test_numpy_input_fn (line 1400) | def _test_numpy_input_fn(self, n_classes):
    method test_binary_classes_numpy_input_fn (line 1436) | def test_binary_classes_numpy_input_fn(self):
    method test_multi_classes_numpy_input_fn (line 1439) | def test_multi_classes_numpy_input_fn(self):
    method _test_pandas_input_fn (line 1442) | def _test_pandas_input_fn(self, n_classes):
    method test_binary_classes_pandas_input_fn (line 1471) | def test_binary_classes_pandas_input_fn(self):
    method test_multi_classes_pandas_input_fn (line 1474) | def test_multi_classes_pandas_input_fn(self):
    method _test_input_fn_from_parse_example (line 1477) | def _test_input_fn_from_parse_example(self, n_classes):
    method test_binary_classes_input_fn_from_parse_example (line 1536) | def test_binary_classes_input_fn_from_parse_example(self):
    method test_multi_classes_input_fn_from_parse_example (line 1539) | def test_multi_classes_input_fn_from_parse_example(self):
  class BaselineLogitFnTest (line 1546) | class BaselineLogitFnTest(tf.test.TestCase):
    method test_basic_logit_correctness (line 1548) | def test_basic_logit_correctness(self):

FILE: tensorflow_estimator/python/estimator/canned/canned_estimator_ds_integration_test.py
  class CannedEstimatorDistributionStrategyTest (line 35) | class CannedEstimatorDistributionStrategyTest(tf.test.TestCase,
    method setUp (line 38) | def setUp(self):
    method dataset_input_fn (line 45) | def dataset_input_fn(self, x, y, batch_size, shuffle):
    method test_canned_estimator (line 69) | def test_canned_estimator(self, distribution, estimator_cls):

FILE: tensorflow_estimator/python/estimator/canned/dnn.py
  function _add_hidden_layer_summary (line 41) | def _add_hidden_layer_summary(value, tag):
  function dnn_logit_fn_builder (line 48) | def dnn_logit_fn_builder(units, hidden_units, feature_columns, activatio...
  function dnn_logit_fn_builder_v2 (line 101) | def dnn_logit_fn_builder_v2(units, hidden_units, feature_columns, activa...
  function _get_previous_name_scope (line 152) | def _get_previous_name_scope():
  class _DNNModel (line 157) | class _DNNModel(tf_keras.Model):
    method __init__ (line 160) | def __init__(self,
    method call (line 232) | def call(self, features, mode):
    method _add_layer (line 259) | def _add_layer(self, layer, layer_name):
  function _name_from_scope_name (line 266) | def _name_from_scope_name(name):
  class _DNNModelV2 (line 278) | class _DNNModelV2(tf_keras.Model):
    method __init__ (line 281) | def __init__(self,
    method call (line 349) | def call(self, features, mode):
  function _validate_features (line 368) | def _validate_features(features):
  function _get_dnn_estimator_spec (line 374) | def _get_dnn_estimator_spec(use_tpu, head, features, labels, mode, logits,
  function _dnn_model_fn (line 393) | def _dnn_model_fn(features,
  function _dnn_model_fn_builder_v2 (line 466) | def _dnn_model_fn_builder_v2(units, hidden_units, feature_columns,
  function dnn_model_fn_v2 (line 513) | def dnn_model_fn_v2(features,
  class DNNClassifierV2 (line 596) | class DNNClassifierV2(estimator.EstimatorV2):
    method __init__ (line 682) | def __init__(
  class DNNClassifier (line 773) | class DNNClassifier(estimator.Estimator):
    method __init__ (line 776) | def __init__(
  class DNNEstimatorV2 (line 821) | class DNNEstimatorV2(estimator.EstimatorV2):
    method __init__ (line 911) | def __init__(self,
  class DNNEstimator (line 976) | class DNNEstimator(estimator.Estimator):
    method __init__ (line 979) | def __init__(self,
  class DNNRegressorV2 (line 1017) | class DNNRegressorV2(estimator.EstimatorV2):
    method __init__ (line 1103) | def __init__(
  class DNNRegressor (line 1186) | class DNNRegressor(estimator.Estimator):
    method __init__ (line 1189) | def __init__(

FILE: tensorflow_estimator/python/estimator/canned/dnn_estimator_test.py
  function _dnn_estimator_fn (line 37) | def _dnn_estimator_fn(weight_column=None, label_dimension=1, **kwargs):
  function _dnn_estimator_classifier_fn (line 48) | def _dnn_estimator_classifier_fn(n_classes=3, **kwargs):
  class DNNLogitFnBuilderTest (line 53) | class DNNLogitFnBuilderTest(tf.test.TestCase):
    method testLongInPy2 (line 55) | def testLongInPy2(self):
  class DNNEstimatorEvaluateTest (line 62) | class DNNEstimatorEvaluateTest(dnn_testing_utils.BaseDNNRegressorEvaluat...
    method __init__ (line 65) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNEstimatorPredictTest (line 71) | class DNNEstimatorPredictTest(dnn_testing_utils.BaseDNNRegressorPredictT...
    method __init__ (line 74) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNEstimatorTrainTest (line 80) | class DNNEstimatorTrainTest(dnn_testing_utils.BaseDNNRegressorTrainTest,
    method __init__ (line 83) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNEstimatorWarmStartingTest (line 89) | class DNNEstimatorWarmStartingTest(dnn_testing_utils.BaseDNNWarmStarting...
    method __init__ (line 92) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNEstimatorIntegrationTest (line 98) | class DNNEstimatorIntegrationTest(tf.test.TestCase):
    method setUp (line 100) | def setUp(self):
    method tearDown (line 103) | def tearDown(self):
    method _test_complete_flow (line 108) | def _test_complete_flow(self,
    method _create_input_fn (line 150) | def _create_input_fn(self, label_dimension, batch_size):
    method test_numpy_input_fn (line 168) | def test_numpy_input_fn(self):
    method test_numpy_input_fn_with_optimizer_instance (line 183) | def test_numpy_input_fn_with_optimizer_instance(self):

FILE: tensorflow_estimator/python/estimator/canned/dnn_linear_combined.py
  function _check_no_sync_replicas_optimizer (line 41) | def _check_no_sync_replicas_optimizer(optimizer):
  function _linear_learning_rate (line 50) | def _linear_learning_rate(num_linear_feature_columns):
  function _add_layer_summary (line 67) | def _add_layer_summary(value, tag):
  function _validate_feature_columns (line 73) | def _validate_feature_columns(linear_feature_columns, dnn_feature_columns):
  function _dnn_linear_combined_model_fn_v2 (line 84) | def _dnn_linear_combined_model_fn_v2(
  function _dnn_linear_combined_model_fn (line 235) | def _dnn_linear_combined_model_fn(features,
  class DNNLinearCombinedClassifierV2 (line 393) | class DNNLinearCombinedClassifierV2(estimator.EstimatorV2):
    method __init__ (line 477) | def __init__(self,
  class DNNLinearCombinedClassifier (line 594) | class DNNLinearCombinedClassifier(estimator.Estimator):
    method __init__ (line 598) | def __init__(self,
  function _init_dnn_linear_combined_estimator (line 651) | def _init_dnn_linear_combined_estimator(head, linear_feature_columns,
  class DNNLinearCombinedEstimatorV2 (line 687) | class DNNLinearCombinedEstimatorV2(estimator.EstimatorV2):
    method __init__ (line 770) | def __init__(self,
  class DNNLinearCombinedEstimator (line 852) | class DNNLinearCombinedEstimator(estimator.Estimator):
    method __init__ (line 855) | def __init__(self,
  class DNNLinearCombinedRegressorV2 (line 899) | class DNNLinearCombinedRegressorV2(estimator.EstimatorV2):
    method __init__ (line 983) | def __init__(self,
  class DNNLinearCombinedRegressor (line 1091) | class DNNLinearCombinedRegressor(estimator.Estimator):
    method __init__ (line 1095) | def __init__(self,

FILE: tensorflow_estimator/python/estimator/canned/dnn_linear_combined_estimator_test.py
  function _dnn_only_estimator_fn (line 37) | def _dnn_only_estimator_fn(hidden_units,
  class DNNOnlyEstimatorEvaluateTest (line 61) | class DNNOnlyEstimatorEvaluateTest(
    method __init__ (line 64) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyEstimatorPredictTest (line 70) | class DNNOnlyEstimatorPredictTest(dnn_testing_utils.BaseDNNRegressorPred...
    method __init__ (line 73) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyEstimatorTrainTest (line 79) | class DNNOnlyEstimatorTrainTest(dnn_testing_utils.BaseDNNRegressorTrainT...
    method __init__ (line 82) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  function _linear_only_estimator_fn (line 88) | def _linear_only_estimator_fn(feature_columns,
  class LinearOnlyEstimatorEvaluateTest (line 108) | class LinearOnlyEstimatorEvaluateTest(
    method __init__ (line 111) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyEstimatorPredictTest (line 117) | class LinearOnlyEstimatorPredictTest(
    method __init__ (line 120) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyEstimatorTrainTest (line 126) | class LinearOnlyEstimatorTrainTest(
    method __init__ (line 129) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNLinearCombinedEstimatorIntegrationTest (line 135) | class DNNLinearCombinedEstimatorIntegrationTest(tf.test.TestCase):
    method setUp (line 137) | def setUp(self):
    method tearDown (line 140) | def tearDown(self):
    method _test_complete_flow (line 145) | def _test_complete_flow(self,
    method _create_input_fn (line 194) | def _create_input_fn(self, label_dimension, batch_size):
    method test_numpy_input_fn (line 212) | def test_numpy_input_fn(self):
    method test_numpy_input_fn_with_optimizer_instance (line 227) | def test_numpy_input_fn_with_optimizer_instance(self):

FILE: tensorflow_estimator/python/estimator/canned/dnn_linear_combined_test.py
  class DNNOnlyModelFnTest (line 53) | class DNNOnlyModelFnTest(dnn_testing_utils.BaseDNNModelFnTest,
    method __init__ (line 56) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
    method _dnn_only_model_fn (line 60) | def _dnn_only_model_fn(self,
  function _linear_regressor_fn (line 86) | def _linear_regressor_fn(feature_columns,
  class LinearOnlyRegressorEvaluationV2Test (line 103) | class LinearOnlyRegressorEvaluationV2Test(
    method __init__ (line 106) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyRegressorPredictV2Test (line 112) | class LinearOnlyRegressorPredictV2Test(
    method __init__ (line 115) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyRegressorIntegrationV2Test (line 121) | class LinearOnlyRegressorIntegrationV2Test(
    method __init__ (line 124) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyRegressorTrainingV2Test (line 130) | class LinearOnlyRegressorTrainingV2Test(
    method __init__ (line 133) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  function _linear_classifier_fn (line 139) | def _linear_classifier_fn(feature_columns,
  class LinearOnlyClassifierTrainingV2Test (line 158) | class LinearOnlyClassifierTrainingV2Test(
    method __init__ (line 161) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyClassifierClassesEvaluationV2Test (line 169) | class LinearOnlyClassifierClassesEvaluationV2Test(
    method __init__ (line 172) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyClassifierPredictV2Test (line 180) | class LinearOnlyClassifierPredictV2Test(
    method __init__ (line 183) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyClassifierIntegrationV2Test (line 191) | class LinearOnlyClassifierIntegrationV2Test(
    method __init__ (line 194) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNLinearCombinedRegressorIntegrationTest (line 203) | class DNNLinearCombinedRegressorIntegrationTest(tf.test.TestCase):
    method setUp (line 205) | def setUp(self):
    method tearDown (line 208) | def tearDown(self):
    method _test_complete_flow_helper (line 213) | def _test_complete_flow_helper(self, linear_feature_columns,
    method _test_complete_flow (line 248) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method _test_complete_flow_dnn_fc_v1 (line 264) | def _test_complete_flow_dnn_fc_v1(self, train_input_fn, eval_input_fn,
    method _test_complete_flow_linear_fc_v1 (line 282) | def _test_complete_flow_linear_fc_v1(self, train_input_fn, eval_input_fn,
    method _test_numpy_input_fn_helper (line 300) | def _test_numpy_input_fn_helper(self, fc_impl, fn_to_run):
    method test_numpy_input_fn_basic (line 327) | def test_numpy_input_fn_basic(self, fc_impl):
    method test_numpy_input_fn_dnn_fc_v1 (line 330) | def test_numpy_input_fn_dnn_fc_v1(self, fc_impl):
    method test_numpy_input_fn_linear_fc_v1 (line 336) | def test_numpy_input_fn_linear_fc_v1(self, fc_impl):
    method _test_pandas_input_fn_helper (line 342) | def _test_pandas_input_fn_helper(self, fc_impl, fn_to_run):
    method test_pandas_input_fn_basic (line 367) | def test_pandas_input_fn_basic(self, fc_impl):
    method test_pandas_input_fn_dnn_fc_v1 (line 370) | def test_pandas_input_fn_dnn_fc_v1(self, fc_impl):
    method test_pandas_input_fn_linear_fc_v1 (line 376) | def test_pandas_input_fn_linear_fc_v1(self, fc_impl):
    method _test_input_fn_from_parse_example_helper (line 382) | def _test_input_fn_from_parse_example_helper(self, fc_impl, fn_to_run):
    method test_input_fn_from_parse_example_basic (line 440) | def test_input_fn_from_parse_example_basic(self, fc_impl):
    method test_input_fn_from_parse_example_dnn_fc_v1 (line 444) | def test_input_fn_from_parse_example_dnn_fc_v1(self, fc_impl):
    method test_input_fn_from_parse_example_linear_fc_v1 (line 450) | def test_input_fn_from_parse_example_linear_fc_v1(self, fc_impl):
  function _dnn_classifier_fn (line 458) | def _dnn_classifier_fn(hidden_units,
  class DNNOnlyClassifierEvaluateV2Test (line 477) | class DNNOnlyClassifierEvaluateV2Test(
    method __init__ (line 480) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyClassifierPredictV2Test (line 486) | class DNNOnlyClassifierPredictV2Test(
    method __init__ (line 489) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyClassifierTrainV2Test (line 495) | class DNNOnlyClassifierTrainV2Test(dnn_testing_utils.BaseDNNClassifierTr...
    method __init__ (line 498) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  function _dnn_regressor_fn (line 505) | def _dnn_regressor_fn(hidden_units,
  class DNNOnlyRegressorEvaluateV2Test (line 522) | class DNNOnlyRegressorEvaluateV2Test(
    method __init__ (line 525) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyRegressorPredictV2Test (line 531) | class DNNOnlyRegressorPredictV2Test(
    method __init__ (line 534) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyRegressorTrainV2Test (line 540) | class DNNOnlyRegressorTrainV2Test(dnn_testing_utils.BaseDNNRegressorTrai...
    method __init__ (line 543) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNLinearCombinedClassifierIntegrationTest (line 550) | class DNNLinearCombinedClassifierIntegrationTest(tf.test.TestCase):
    method setUp (line 552) | def setUp(self):
    method tearDown (line 555) | def tearDown(self):
    method _as_label (line 560) | def _as_label(self, data_in_float):
    method _test_complete_flow (line 563) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method test_numpy_input_fn (line 603) | def test_numpy_input_fn(self, fc_impl):
    method test_pandas_input_fn (line 633) | def test_pandas_input_fn(self, fc_impl):
    method test_input_fn_from_parse_example (line 659) | def test_input_fn_from_parse_example(self, fc_impl):
  class DNNLinearCombinedTests (line 722) | class DNNLinearCombinedTests(tf.test.TestCase):
    method setUp (line 724) | def setUp(self):
    method tearDown (line 727) | def tearDown(self):
    method test_train_op_calls_both_dnn_and_linear (line 731) | def test_train_op_calls_both_dnn_and_linear(self, fc_impl):
    method test_dnn_and_linear_logits_are_added (line 757) | def test_dnn_and_linear_logits_are_added(self, fc_impl):
  class DNNLinearCombinedWarmStartingTest (line 785) | class DNNLinearCombinedWarmStartingTest(tf.test.TestCase):
    method setUp (line 787) | def setUp(self):
    method tearDown (line 801) | def tearDown(self):
    method test_classifier_basic_warm_starting (line 806) | def test_classifier_basic_warm_starting(self, fc_impl):
    method test_regressor_basic_warm_starting (line 856) | def test_regressor_basic_warm_starting(self, fc_impl):
    method test_warm_starting_selective_variables (line 904) | def test_warm_starting_selective_variables(self, fc_impl):

FILE: tensorflow_estimator/python/estimator/canned/dnn_test_fc_v2.py
  function _dnn_classifier_fn (line 52) | def _dnn_classifier_fn(*args, **kwargs):
  class DNNModelFnV2Test (line 56) | class DNNModelFnV2Test(dnn_testing_utils.BaseDNNModelFnTest, tf.test.Tes...
    method __init__ (line 58) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNLogitFnV2Test (line 64) | class DNNLogitFnV2Test(dnn_testing_utils.BaseDNNLogitFnTest, tf.test.Tes...
    method __init__ (line 66) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNWarmStartingV2Test (line 72) | class DNNWarmStartingV2Test(dnn_testing_utils.BaseDNNWarmStartingTest,
    method __init__ (line 75) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNClassifierEvaluateV2Test (line 81) | class DNNClassifierEvaluateV2Test(
    method __init__ (line 84) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNClassifierPredictV2Test (line 90) | class DNNClassifierPredictV2Test(dnn_testing_utils.BaseDNNClassifierPred...
    method __init__ (line 93) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNClassifierTrainV2Test (line 99) | class DNNClassifierTrainV2Test(dnn_testing_utils.BaseDNNClassifierTrainT...
    method __init__ (line 102) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  function _dnn_regressor_fn (line 108) | def _dnn_regressor_fn(*args, **kwargs):
  class DNNRegressorEvaluateV2Test (line 112) | class DNNRegressorEvaluateV2Test(dnn_testing_utils.BaseDNNRegressorEvalu...
    method __init__ (line 115) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNRegressorPredictV2Test (line 121) | class DNNRegressorPredictV2Test(dnn_testing_utils.BaseDNNRegressorPredic...
    method __init__ (line 124) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNRegressorTrainV2Test (line 130) | class DNNRegressorTrainV2Test(dnn_testing_utils.BaseDNNRegressorTrainTest,
    method __init__ (line 133) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  function _queue_parsed_features (line 139) | def _queue_parsed_features(feature_map):
  class DNNRegressorIntegrationTest (line 154) | class DNNRegressorIntegrationTest(tf.test.TestCase, parameterized.TestCa...
    method setUp (line 156) | def setUp(self):
    method tearDown (line 159) | def tearDown(self):
    method _test_complete_flow (line 164) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method test_numpy_input_fn (line 200) | def test_numpy_input_fn(self):
    method test_pandas_input_fn (line 226) | def test_pandas_input_fn(self):
    method test_input_fn_from_parse_example (line 250) | def test_input_fn_from_parse_example(self):
  class DNNClassifierIntegrationTest (line 308) | class DNNClassifierIntegrationTest(tf.test.TestCase):
    method setUp (line 310) | def setUp(self):
    method tearDown (line 313) | def tearDown(self):
    method _as_label (line 318) | def _as_label(self, data_in_float):
    method _test_complete_flow (line 321) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method test_numpy_input_fn (line 357) | def test_numpy_input_fn(self):
    method test_pandas_input_fn (line 386) | def test_pandas_input_fn(self):
    method test_input_fn_from_parse_example (line 411) | def test_input_fn_from_parse_example(self):
  class DNNTrainingMode (line 472) | class DNNTrainingMode(tf.test.TestCase):
    method setUp (line 475) | def setUp(self):
    method tearDown (line 480) | def tearDown(self):
    method _create_data (line 485) | def _create_data(self):
    method _get_estimator (line 490) | def _get_estimator(self):
    method test_train_vs_eval_mode (line 500) | def test_train_vs_eval_mode(self):

FILE: tensorflow_estimator/python/estimator/canned/dnn_testing_utils.py
  function assert_close (line 59) | def assert_close(expected, actual, rtol=1e-04, message='', name='assert_...
  function create_checkpoint (line 75) | def create_checkpoint(weights_and_biases,
  function mock_head (line 123) | def mock_head(testcase, hidden_units, logits_dimension, expected_logits):
  function mock_optimizer (line 191) | def mock_optimizer(testcase, hidden_units, expected_loss=None):
  class BaseDNNModelFnTest (line 242) | class BaseDNNModelFnTest(object):
    method __init__ (line 245) | def __init__(self, dnn_model_fn, fc_impl=feature_column_v2):
    method setUp (line 249) | def setUp(self):
    method tearDown (line 252) | def tearDown(self):
    method _test_logits (line 257) | def _test_logits(self, mode, hidden_units, logits_dimension, inputs,
    method test_one_dim_logits (line 289) | def test_one_dim_logits(self):
    method test_multi_dim_logits (line 313) | def test_multi_dim_logits(self):
    method test_multi_example_multi_dim_logits (line 338) | def test_multi_example_multi_dim_logits(self):
    method test_multi_dim_input_one_dim_logits (line 367) | def test_multi_dim_input_one_dim_logits(self):
    method test_multi_dim_input_multi_dim_logits (line 392) | def test_multi_dim_input_multi_dim_logits(self):
    method test_multi_feature_column_multi_dim_logits (line 417) | def test_multi_feature_column_multi_dim_logits(self):
    method test_multi_feature_column_mix_multi_dim_logits (line 468) | def test_multi_feature_column_mix_multi_dim_logits(self):
    method test_features_tensor_raises_value_error (line 519) | def test_features_tensor_raises_value_error(self):
  class BaseDNNLogitFnTest (line 547) | class BaseDNNLogitFnTest(object):
    method __init__ (line 550) | def __init__(self, dnn_logit_fn_builder, fc_impl=feature_column_v2):
    method setUp (line 554) | def setUp(self):
    method tearDown (line 557) | def tearDown(self):
    method _test_logits (line 562) | def _test_logits(self,
    method test_one_dim_logits (line 589) | def test_one_dim_logits(self):
    method test_one_dim_logits_with_batch_norm (line 612) | def test_one_dim_logits_with_batch_norm(self):
    method test_multi_dim_logits (line 677) | def test_multi_dim_logits(self):
    method test_multi_example_multi_dim_logits (line 701) | def test_multi_example_multi_dim_logits(self):
    method test_multi_dim_input_one_dim_logits (line 729) | def test_multi_dim_input_one_dim_logits(self):
    method test_multi_dim_input_multi_dim_logits (line 754) | def test_multi_dim_input_multi_dim_logits(self):
    method test_multi_feature_column_multi_dim_logits (line 778) | def test_multi_feature_column_multi_dim_logits(self):
    method test_multi_feature_column_mix_multi_dim_logits (line 822) | def test_multi_feature_column_mix_multi_dim_logits(self):
  class BaseDNNWarmStartingTest (line 867) | class BaseDNNWarmStartingTest(object):
    method __init__ (line 869) | def __init__(self,
    method setUp (line 877) | def setUp(self):
    method tearDown (line 895) | def tearDown(self):
    method assertAllNotClose (line 900) | def assertAllNotClose(self, t1, t2):
    method test_classifier_basic_warm_starting (line 911) | def test_classifier_basic_warm_starting(self):
    method test_regressor_basic_warm_starting (line 949) | def test_regressor_basic_warm_starting(self):
    method test_warm_starting_selective_variables (line 985) | def test_warm_starting_selective_variables(self):
    method test_warm_starting_with_vocab_remapping (line 1034) | def test_warm_starting_with_vocab_remapping(self):
    method test_warm_starting_with_naming_change (line 1128) | def test_warm_starting_with_naming_change(self):
  class BaseDNNClassifierEvaluateTest (line 1183) | class BaseDNNClassifierEvaluateTest(object):
    method __init__ (line 1185) | def __init__(self, dnn_classifier_fn, fc_impl=feature_column_v2):
    method setUp (line 1189) | def setUp(self):
    method tearDown (line 1192) | def tearDown(self):
    method test_one_dim (line 1197) | def test_one_dim(self):
    method test_multi_dim (line 1250) | def test_multi_dim(self):
    method test_float_labels (line 1287) | def test_float_labels(self):
    method test_multi_dim_weights (line 1315) | def test_multi_dim_weights(self):
  class BaseDNNRegressorEvaluateTest (line 1346) | class BaseDNNRegressorEvaluateTest(object):
    method __init__ (line 1348) | def __init__(self, dnn_regressor_fn, fc_impl=feature_column_v2):
    method setUp (line 1352) | def setUp(self):
    method tearDown (line 1355) | def tearDown(self):
    method test_one_dim (line 1360) | def test_one_dim(self):
    method test_multi_dim (line 1392) | def test_multi_dim(self):
    method test_multi_dim_weights (line 1428) | def test_multi_dim_weights(self):
  class BaseDNNClassifierPredictTest (line 1458) | class BaseDNNClassifierPredictTest(object):
    method __init__ (line 1460) | def __init__(self, dnn_classifier_fn, fc_impl=feature_column_v2):
    method setUp (line 1464) | def setUp(self):
    method tearDown (line 1467) | def tearDown(self):
    method _test_one_dim (line 1472) | def _test_one_dim(self, label_vocabulary, label_output_fn):
    method test_one_dim_without_label_vocabulary (line 1513) | def test_one_dim_without_label_vocabulary(self):
    method test_one_dim_with_label_vocabulary (line 1517) | def test_one_dim_with_label_vocabulary(self):
    method _test_multi_dim_with_3_classes (line 1523) | def _test_multi_dim_with_3_classes(self, label_vocabulary, label_outpu...
    method test_multi_dim_with_3_classes_but_no_label_vocab (line 1577) | def test_multi_dim_with_3_classes_but_no_label_vocab(self):
    method test_multi_dim_with_3_classes_and_label_vocab (line 1581) | def test_multi_dim_with_3_classes_and_label_vocab(self):
  class BaseDNNRegressorPredictTest (line 1588) | class BaseDNNRegressorPredictTest(object):
    method __init__ (line 1590) | def __init__(self, dnn_regressor_fn, fc_impl=feature_column_v2):
    method setUp (line 1594) | def setUp(self):
    method tearDown (line 1597) | def tearDown(self):
    method test_one_dim (line 1602) | def test_one_dim(self):
    method test_multi_dim (line 1626) | def test_multi_dim(self):
  class _SummaryHook (line 1655) | class _SummaryHook(tf.compat.v1.train.SessionRunHook):
    method __init__ (line 1658) | def __init__(self):
    method begin (line 1661) | def begin(self):
    method before_run (line 1664) | def before_run(self, run_context):
    method after_run (line 1667) | def after_run(self, run_context, run_values):
    method summaries (line 1672) | def summaries(self):
  function _assert_checkpoint (line 1676) | def _assert_checkpoint(testcase, global_step, input_units, hidden_units,
  function _assert_simple_summary (line 1712) | def _assert_simple_summary(testcase, expected_values, actual_summary):
  class BaseDNNClassifierTrainTest (line 1728) | class BaseDNNClassifierTrainTest(object):
    method __init__ (line 1730) | def __init__(self, dnn_classifier_fn, fc_impl=feature_column_v2):
    method setUp (line 1734) | def setUp(self):
    method tearDown (line 1737) | def tearDown(self):
    method test_from_scratch_with_default_optimizer_binary (line 1742) | def test_from_scratch_with_default_optimizer_binary(self):
    method test_from_scratch_with_default_optimizer_multi_class (line 1763) | def test_from_scratch_with_default_optimizer_multi_class(self):
    method test_from_scratch_validate_summary (line 1786) | def test_from_scratch_validate_summary(self):
    method test_binary_classification (line 1820) | def test_binary_classification(self):
    method test_binary_classification_float_labels (line 1872) | def test_binary_classification_float_labels(self):
    method test_multi_class (line 1904) | def test_multi_class(self):
  class BaseDNNRegressorTrainTest (line 1959) | class BaseDNNRegressorTrainTest(object):
    method __init__ (line 1961) | def __init__(self, dnn_regressor_fn, fc_impl=feature_column_v2):
    method setUp (line 1965) | def setUp(self):
    method tearDown (line 1968) | def tearDown(self):
    method test_from_scratch_with_default_optimizer (line 1973) | def test_from_scratch_with_default_optimizer(self):
    method test_from_scratch (line 1994) | def test_from_scratch(self):
    method test_one_dim (line 2028) | def test_one_dim(self):
    method test_multi_dim (line 2081) | def test_multi_dim(self):

FILE: tensorflow_estimator/python/estimator/canned/head.py
  function _summary_key (line 55) | def _summary_key(head_name, val):
  function _create_eval_metrics_tuple (line 59) | def _create_eval_metrics_tuple(fn, kwargs):
  class _Head (line 90) | class _Head(object):
    method name (line 147) | def name(self):
    method logits_dimension (line 156) | def logits_dimension(self):
    method create_loss (line 167) | def create_loss(self, features, mode, logits, labels):
    method create_estimator_spec (line 195) | def create_estimator_spec(self,
    method _create_tpu_estimator_spec (line 242) | def _create_tpu_estimator_spec(self,
  function _check_dense_labels_match_logits_and_reshape (line 276) | def _check_dense_labels_match_logits_and_reshape(labels, logits,
  function _get_weights_and_check_match_logits (line 355) | def _get_weights_and_check_match_logits(features,
  function _check_logits_final_dim (line 444) | def _check_logits_final_dim(logits, expected_logits_dimension):
  function _validate_loss_fn_args (line 473) | def _validate_loss_fn_args(loss_fn):
  function _validate_n_classes (line 495) | def _validate_n_classes(n_classes):
  function _call_loss_fn (line 521) | def _call_loss_fn(loss_fn, labels, logits, features, expected_loss_dim=1):
  function _indicator_labels_mean (line 560) | def _indicator_labels_mean(labels, weights=None, name=None):
  function _all_class_ids (line 568) | def _all_class_ids(logits, n_classes):
  function _all_classes (line 576) | def _all_classes(logits, n_classes, label_vocabulary=None):
  function _classification_output (line 587) | def _classification_output(scores, n_classes, label_vocabulary=None):
  function _accuracy_baseline (line 602) | def _accuracy_baseline(labels_mean):
  function _predictions_mean (line 619) | def _predictions_mean(predictions, weights=None, name=None):
  function _auc (line 629) | def _auc(labels, predictions, weights=None, curve='ROC', name=None):
  function _accuracy_at_threshold (line 643) | def _accuracy_at_threshold(labels, predictions, weights, threshold, name...
  function _precision_at_threshold (line 655) | def _precision_at_threshold(labels, predictions, weights, threshold, nam...
  function _recall_at_threshold (line 668) | def _recall_at_threshold(labels, predictions, weights, threshold, name=N...
  function _multi_class_head_with_softmax_cross_entropy_loss (line 681) | def _multi_class_head_with_softmax_cross_entropy_loss(
  class _MultiClassHeadWithSoftmaxCrossEntropyLoss (line 755) | class _MultiClassHeadWithSoftmaxCrossEntropyLoss(_Head):
    method __init__ (line 758) | def __init__(self,
    method name (line 775) | def name(self):
    method logits_dimension (line 779) | def logits_dimension(self):
    method _eval_metric_ops (line 782) | def _eval_metric_ops(self, labels, class_ids, weights, unreduced_loss,
    method _label_ids (line 808) | def _label_ids(self, labels):
    method create_loss (line 825) | def create_loss(self, features, mode, logits, labels):
    method _create_tpu_estimator_spec (line 856) | def _create_tpu_estimator_spec(self,
  function _binary_logistic_head_with_sigmoid_cross_entropy_loss (line 999) | def _binary_logistic_head_with_sigmoid_cross_entropy_loss(
  class _BinaryLogisticHeadWithSigmoidCrossEntropyLoss (line 1084) | class _BinaryLogisticHeadWithSigmoidCrossEntropyLoss(_Head):
    method __init__ (line 1087) | def __init__(self,
    method name (line 1102) | def name(self):
    method logits_dimension (line 1106) | def logits_dimension(self):
    method _eval_metric_ops (line 1109) | def _eval_metric_ops(self, labels, logits, logistic, class_ids, weights,
    method create_loss (line 1195) | def create_loss(self, features, mode, logits, labels):
    method _create_tpu_estimator_spec (line 1227) | def _create_tpu_estimator_spec(self,
  function _regression_head (line 1378) | def _regression_head(weight_column=None,
  class _RegressionHeadWithMeanSquaredErrorLoss (line 1447) | class _RegressionHeadWithMeanSquaredErrorLoss(_Head):
    method __init__ (line 1450) | def __init__(self,
    method name (line 1468) | def name(self):
    method logits_dimension (line 1472) | def logits_dimension(self):
    method create_loss (line 1475) | def create_loss(self, features, mode, logits, labels):
    method _eval_metric_ops (line 1509) | def _eval_metric_ops(self, predicted_value, labels, weights, unreduced...
    method _create_tpu_estimator_spec (line 1532) | def _create_tpu_estimator_spec(self,
  function _append_update_ops (line 1657) | def _append_update_ops(train_op):
  function _assert_range (line 1665) | def _assert_range(labels, n_classes, message=None):
  function _binary_logistic_or_multi_class_head (line 1677) | def _binary_logistic_or_multi_class_head(n_classes, weight_column,

FILE: tensorflow_estimator/python/estimator/canned/head_test.py
  function _initialize_variables (line 36) | def _initialize_variables(test_case, scaffold):
  function _assert_simple_summaries (line 47) | def _assert_simple_summaries(test_case,
  function _assert_no_hooks (line 67) | def _assert_no_hooks(test_case, spec):
  function _sigmoid (line 72) | def _sigmoid(logits):
  class CreateEstimatorSpecTest (line 77) | class CreateEstimatorSpecTest(tf.test.TestCase):
    class _HeadWithTPUSupport (line 79) | class _HeadWithTPUSupport(head_lib._Head):
      method name (line 82) | def name(self):
      method logits_dimension (line 85) | def logits_dimension(self):
      method create_loss (line 88) | def create_loss(self, features, mode, logits, labels):
      method _create_tpu_estimator_spec (line 91) | def _create_tpu_estimator_spec(self,
    class _HeadWithOutTPUSupport (line 102) | class _HeadWithOutTPUSupport(head_lib._Head):
      method name (line 105) | def name(self):
      method logits_dimension (line 108) | def logits_dimension(self):
      method create_loss (line 111) | def create_loss(self, features, mode, logits, labels):
      method create_estimator_spec (line 114) | def create_estimator_spec(self,
    class _InvalidHead (line 125) | class _InvalidHead(head_lib._Head):
      method name (line 128) | def name(self):
      method logits_dimension (line 131) | def logits_dimension(self):
      method create_loss (line 134) | def create_loss(self, features, mode, logits, labels):
    method test_head_override_tpu_estimator_spec (line 137) | def test_head_override_tpu_estimator_spec(self):
    method test_head_override_estimator_spec (line 147) | def test_head_override_estimator_spec(self):
    method test_invalid_head_class (line 158) | def test_invalid_head_class(self):
  class MultiClassHeadWithSoftmaxCrossEntropyLoss (line 173) | class MultiClassHeadWithSoftmaxCrossEntropyLoss(tf.test.TestCase):
    method setUp (line 175) | def setUp(self):
    method test_n_classes_is_none (line 178) | def test_n_classes_is_none(self):
    method test_n_classes_is_2 (line 182) | def test_n_classes_is_2(self):
    method test_invalid_loss_reduction (line 186) | def test_invalid_loss_reduction(self):
    method test_loss_fn_arg_labels_missing (line 195) | def test_loss_fn_arg_labels_missing(self):
    method test_loss_fn_arg_logits_missing (line 206) | def test_loss_fn_arg_logits_missing(self):
    method test_loss_fn_arg_features_ok (line 217) | def test_loss_fn_arg_features_ok(self):
    method test_loss_fn_arg_invalid (line 225) | def test_loss_fn_arg_invalid(self):
    method test_invalid_logits_shape (line 235) | def test_invalid_logits_shape(self):
    method test_invalid_labels_shape (line 261) | def test_invalid_labels_shape(self):
    method test_invalid_labels_type (line 297) | def test_invalid_labels_type(self):
    method test_invalid_labels_values (line 326) | def test_invalid_labels_values(self):
    method test_invalid_labels_sparse_tensor (line 356) | def test_invalid_labels_sparse_tensor(self):
    method test_incompatible_labels_shape (line 375) | def test_incompatible_labels_shape(self):
    method test_name (line 416) | def test_name(self):
    method test_predict (line 421) | def test_predict(self):
    method test_predict_with_tensor_n_classes (line 471) | def test_predict_with_tensor_n_classes(self):
    method test_predict_with_vocabulary_list (line 521) | def test_predict_with_vocabulary_list(self):
    method test_weight_should_not_impact_prediction (line 544) | def test_weight_should_not_impact_prediction(self):
    method test_eval_create_loss (line 570) | def test_eval_create_loss(self):
    method test_eval_create_loss_loss_fn (line 587) | def test_eval_create_loss_loss_fn(self):
    method test_eval_create_loss_loss_fn_wrong_shape (line 613) | def test_eval_create_loss_loss_fn_wrong_shape(self):
    method test_eval_labels_none (line 639) | def test_eval_labels_none(self):
    method test_eval (line 652) | def test_eval(self):
    method test_eval_metric_ops_with_head_name (line 693) | def test_eval_metric_ops_with_head_name(self):
    method test_eval_with_regularization_losses (line 710) | def test_eval_with_regularization_losses(self):
    method test_eval_with_label_vocabulary_create_loss (line 756) | def test_eval_with_label_vocabulary_create_loss(self):
    method test_eval_with_label_vocabulary (line 772) | def test_eval_with_label_vocabulary(self):
    method test_weighted_multi_example_eval (line 805) | def test_weighted_multi_example_eval(self):
    method test_train_create_loss (line 856) | def test_train_create_loss(self):
    method test_train_create_loss_loss_reduction (line 881) | def test_train_create_loss_loss_reduction(self):
    method test_train_labels_none (line 908) | def test_train_labels_none(self):
    method test_train (line 926) | def test_train(self):
    method test_train_with_optimizer (line 973) | def test_train_with_optimizer(self):
    method test_train_with_update_ops (line 1009) | def test_train_with_update_ops(self):
    method test_train_summaries_with_head_name (line 1040) | def test_train_summaries_with_head_name(self):
    method test_train_with_regularization_losses (line 1077) | def test_train_with_regularization_losses(self):
    method test_train_one_dim_create_loss (line 1127) | def test_train_one_dim_create_loss(self):
    method test_train_one_dim (line 1160) | def test_train_one_dim(self):
    method test_train_with_vocabulary_create_loss (line 1219) | def test_train_with_vocabulary_create_loss(self):
    method test_train_with_vocabulary (line 1236) | def test_train_with_vocabulary(self):
    method test_weighted_multi_example_train (line 1264) | def test_weighted_multi_example_train(self):
    method test_multi_dim_weighted_train_create_loss (line 1324) | def test_multi_dim_weighted_train_create_loss(self):
    method test_multi_dim_weighted_train (line 1354) | def test_multi_dim_weighted_train(self):
    method test_multi_dim_train_weights_wrong_inner_dim (line 1391) | def test_multi_dim_train_weights_wrong_inner_dim(self):
    method test_multi_dim_train_weights_wrong_outer_dim (line 1417) | def test_multi_dim_train_weights_wrong_outer_dim(self):
    method test_multi_dim_weighted_eval (line 1445) | def test_multi_dim_weighted_eval(self):
  class BinaryLogisticHeadWithSigmoidCrossEntropyLossTest (line 1488) | class BinaryLogisticHeadWithSigmoidCrossEntropyLossTest(tf.test.TestCase):
    method setUp (line 1490) | def setUp(self):
    method test_threshold_too_small (line 1493) | def test_threshold_too_small(self):
    method test_threshold_too_large (line 1498) | def test_threshold_too_large(self):
    method test_invalid_loss_reduction (line 1503) | def test_invalid_loss_reduction(self):
    method test_loss_fn_arg_labels_missing (line 1512) | def test_loss_fn_arg_labels_missing(self):
    method test_loss_fn_arg_logits_missing (line 1523) | def test_loss_fn_arg_logits_missing(self):
    method test_loss_fn_arg_features_ok (line 1534) | def test_loss_fn_arg_features_ok(self):
    method test_loss_fn_arg_invalid (line 1541) | def test_loss_fn_arg_invalid(self):
    method test_invalid_logits_shape (line 1551) | def test_invalid_logits_shape(self):
    method test_invalid_labels_shape (line 1576) | def test_invalid_labels_shape(self):
    method test_incompatible_labels_shape (line 1609) | def test_incompatible_labels_shape(self):
    method test_name (line 1658) | def test_name(self):
    method test_predict (line 1663) | def test_predict(self):
    method test_predict_with_vocabulary_list (line 1721) | def test_predict_with_vocabulary_list(self):
    method test_eval_create_loss (line 1739) | def test_eval_create_loss(self):
    method test_eval_labels_none (line 1755) | def test_eval_labels_none(self):
    method test_eval (line 1767) | def test_eval(self):
    method test_eval_metric_ops_with_head_name (line 1812) | def test_eval_metric_ops_with_head_name(self):
    method test_eval_with_regularization_losses (line 1835) | def test_eval_with_regularization_losses(self):
    method test_eval_with_vocabulary_list_create_loss (line 1884) | def test_eval_with_vocabulary_list_create_loss(self):
    method test_eval_with_vocabulary_list (line 1897) | def test_eval_with_vocabulary_list(self):
    method test_eval_with_thresholds_create_loss (line 1916) | def test_eval_with_thresholds_create_loss(self):
    method test_eval_with_thresholds (line 1937) | def test_eval_with_thresholds(self):
    method test_train_create_loss (line 1990) | def test_train_create_loss(self):
    method test_train_create_loss_loss_reduction (line 2011) | def test_train_create_loss_loss_reduction(self):
    method test_eval_create_loss_loss_fn (line 2034) | def test_eval_create_loss_loss_fn(self):
    method test_eval_create_loss_loss_fn_wrong_shape (line 2060) | def test_eval_create_loss_loss_fn_wrong_shape(self):
    method test_train_labels_none (line 2086) | def test_train_labels_none(self):
    method test_train (line 2103) | def test_train(self):
    method test_train_with_optimizer (line 2152) | def test_train_with_optimizer(self):
    method test_train_with_update_ops (line 2186) | def test_train_with_update_ops(self):
    method test_train_summaries_with_head_name (line 2216) | def test_train_summaries_with_head_name(self):
    method test_train_with_regularization_losses (line 2253) | def test_train_with_regularization_losses(self):
    method test_float_labels_invalid_values (line 2300) | def test_float_labels_invalid_values(self):
    method test_float_labels_train_create_loss (line 2312) | def test_float_labels_train_create_loss(self):
    method test_float_labels_train (line 2333) | def test_float_labels_train(self):
    method test_float_labels_eval_create_loss (line 2368) | def test_float_labels_eval_create_loss(self):
    method test_float_labels_eval (line 2389) | def test_float_labels_eval(self):
    method test_weighted_multi_example_predict (line 2416) | def test_weighted_multi_example_predict(self):
    method test_weighted_multi_example_eval (line 2449) | def test_weighted_multi_example_eval(self):
    method test_train_one_dim_create_loss (line 2505) | def test_train_one_dim_create_loss(self):
    method test_train_one_dim (line 2538) | def test_train_one_dim(self):
    method test_weighted_multi_example_train (line 2595) | def test_weighted_multi_example_train(self):
    method test_multi_dim_weighted_train_create_loss (line 2649) | def test_multi_dim_weighted_train_create_loss(self):
    method test_multi_dim_weighted_train (line 2678) | def test_multi_dim_weighted_train(self):
    method test_multi_dim_train_weights_wrong_inner_dim (line 2715) | def test_multi_dim_train_weights_wrong_inner_dim(self):
    method test_multi_dim_train_weights_wrong_outer_dim (line 2741) | def test_multi_dim_train_weights_wrong_outer_dim(self):
    method test_multi_dim_weighted_eval (line 2770) | def test_multi_dim_weighted_eval(self):
  class RegressionHead (line 2824) | class RegressionHead(tf.test.TestCase):
    method setUp (line 2826) | def setUp(self):
    method test_invalid_label_dimension (line 2829) | def test_invalid_label_dimension(self):
    method test_invalid_loss_reduction (line 2835) | def test_invalid_loss_reduction(self):
    method test_loss_fn_arg_labels_missing (line 2843) | def test_loss_fn_arg_labels_missing(self):
    method test_loss_fn_arg_logits_missing (line 2853) | def test_loss_fn_arg_logits_missing(self):
    method test_loss_fn_arg_features_ok (line 2863) | def test_loss_fn_arg_features_ok(self):
    method test_loss_fn_arg_invalid (line 2869) | def test_loss_fn_arg_invalid(self):
    method test_invalid_logits (line 2878) | def test_invalid_logits(self):
    method test_incompatible_labels_eval (line 2901) | def test_incompatible_labels_eval(self):
    method test_incompatible_labels_train (line 2950) | def test_incompatible_labels_train(self):
    method test_name (line 3001) | def test_name(self):
    method test_predict (line 3005) | def test_predict(self):
    method test_predict_with_inverse_link_fn (line 3039) | def test_predict_with_inverse_link_fn(self):
    method test_eval_create_loss (line 3081) | def test_eval_create_loss(self):
    method test_eval_create_loss_loss_fn (line 3094) | def test_eval_create_loss_loss_fn(self):
    method test_eval_create_loss_loss_fn_wrong_shape (line 3119) | def test_eval_create_loss_loss_fn_wrong_shape(self):
    method test_eval_labels_none (line 3144) | def test_eval_labels_none(self):
    method test_eval (line 3156) | def test_eval(self):
    method test_eval_metric_ops_with_head_name_for_regression (line 3197) | def test_eval_metric_ops_with_head_name_for_regression(self):
    method test_eval_with_regularization_losses (line 3214) | def test_eval_with_regularization_losses(self):
    method test_train_create_loss (line 3261) | def test_train_create_loss(self):
    method test_train_create_loss_loss_reduction (line 3281) | def test_train_create_loss_loss_reduction(self):
    method test_train_labels_none (line 3303) | def test_train_labels_none(self):
    method test_train (line 3320) | def test_train(self):
    method test_train_with_optimizer (line 3375) | def test_train_with_optimizer(self):
    method test_train_with_update_ops (line 3410) | def test_train_with_update_ops(self):
    method test_train_summaries_with_head_name (line 3440) | def test_train_summaries_with_head_name(self):
    method test_train_with_regularization_losses (line 3479) | def test_train_with_regularization_losses(self):
    method test_weighted_multi_example_eval (line 3529) | def test_weighted_multi_example_eval(self):
    method test_weight_with_numeric_column (line 3575) | def test_weight_with_numeric_column(self):
    method test_weighted_multi_example_train (line 3601) | def test_weighted_multi_example_train(self):
    method test_train_one_dim_create_loss (line 3658) | def test_train_one_dim_create_loss(self):
    method test_train_one_dim (line 3684) | def test_train_one_dim(self):
    method test_weighted_multi_value_eval_create_loss (line 3746) | def test_weighted_multi_value_eval_create_loss(self):
    method test_weighted_multi_value_eval (line 3765) | def test_weighted_multi_value_eval(self):
    method test_weighted_multi_value_train_create_loss (line 3811) | def test_weighted_multi_value_train_create_loss(self):
    method test_weighted_multi_value_train (line 3830) | def test_weighted_multi_value_train(self):
    method test_weighted_multi_batch_eval (line 3890) | def test_weighted_multi_batch_eval(self):
    method test_weighted_multi_batch_train (line 3962) | def test_weighted_multi_batch_train(self):
    method test_multi_dim_weighted_train_create_loss (line 4009) | def test_multi_dim_weighted_train_create_loss(self):
    method test_multi_dim_weighted_train (line 4040) | def test_multi_dim_weighted_train(self):
    method test_multi_dim_train_weights_wrong_inner_dim (line 4073) | def test_multi_dim_train_weights_wrong_inner_dim(self):
    method test_multi_dim_train_weights_wrong_outer_dim (line 4102) | def test_multi_dim_train_weights_wrong_outer_dim(self):

FILE: tensorflow_estimator/python/estimator/canned/kmeans.py
  class _LossRelativeChangeHook (line 36) | class _LossRelativeChangeHook(tf.compat.v1.train.SessionRunHook):
    method __init__ (line 39) | def __init__(self, loss_tensor, tolerance):
    method before_run (line 50) | def before_run(self, run_context):
    method after_run (line 54) | def after_run(self, run_context, run_values):
  class _InitializeClustersHook (line 65) | class _InitializeClustersHook(tf.compat.v1.train.SessionRunHook):
    method __init__ (line 72) | def __init__(self, init_op, is_initialized_var, is_chief):
    method after_create_session (line 86) | def after_create_session(self, session, coord):
  function _parse_features_if_necessary (line 102) | def _parse_features_if_necessary(features, feature_columns):
  class _ModelFn (line 130) | class _ModelFn(object):
    method __init__ (line 133) | def __init__(self, num_clusters, initial_clusters, distance_metric, seed,
    method model_fn (line 147) | def model_fn(self, features, mode, config):
  class KMeansClustering (line 241) | class KMeansClustering(estimator.Estimator):
    method __init__ (line 315) | def __init__(self,
    method _predict_one_key (line 423) | def _predict_one_key(self, input_fn, predict_key):
    method predict_cluster_index (line 427) | def predict_cluster_index(self, input_fn):
    method score (line 440) | def score(self, input_fn):
    method transform (line 456) | def transform(self, input_fn):
    method cluster_centers (line 477) | def cluster_centers(self):

FILE: tensorflow_estimator/python/estimator/canned/kmeans_test.py
  function normalize (line 36) | def normalize(x):
  function cosine_similarity (line 40) | def cosine_similarity(x, y):
  function make_random_centers (line 44) | def make_random_centers(num_centers, num_dims, center_norm=500):
  function make_random_points (line 49) | def make_random_points(centers, num_points, max_offset=20):
  class KMeansTestBase (line 58) | class KMeansTestBase(tf.test.TestCase):
    method input_fn (line 60) | def input_fn(self,
    method config (line 106) | def config(tf_random_seed):
    method initial_clusters (line 110) | def initial_clusters(self):
    method batch_size (line 114) | def batch_size(self):
    method use_mini_batch (line 118) | def use_mini_batch(self):
    method mini_batch_steps_per_iteration (line 122) | def mini_batch_steps_per_iteration(self):
  class KMeansTest (line 127) | class KMeansTest(KMeansTestBase):
    method setUp (line 129) | def setUp(self):
    method _kmeans (line 139) | def _kmeans(self, relative_tolerance=None):
    method test_clusters (line 149) | def test_clusters(self):
    method test_fit (line 155) | def test_fit(self):
    method test_monitor (line 165) | def test_monitor(self):
    method _infer_helper (line 186) | def _infer_helper(self, kmeans, clusters, num_points):
    method test_infer (line 208) | def test_infer(self):
    method _parse_feature_dict_helper (line 219) | def _parse_feature_dict_helper(self, features, parsed_feature_dict):
    method test_parse_features (line 228) | def test_parse_features(self):
  class KMeansTestMultiStageInit (line 262) | class KMeansTestMultiStageInit(KMeansTestBase):
    method test_random (line 264) | def test_random(self):
    method test_kmeans_plus_plus_batch_just_right (line 281) | def test_kmeans_plus_plus_batch_just_right(self):
    method test_kmeans_plus_plus_batch_too_small (line 297) | def test_kmeans_plus_plus_batch_too_small(self):
  class MiniBatchKMeansTest (line 315) | class MiniBatchKMeansTest(KMeansTest):
    method batch_size (line 318) | def batch_size(self):
    method use_mini_batch (line 322) | def use_mini_batch(self):
  class FullBatchAsyncKMeansTest (line 327) | class FullBatchAsyncKMeansTest(KMeansTest):
    method batch_size (line 330) | def batch_size(self):
    method use_mini_batch (line 334) | def use_mini_batch(self):
    method mini_batch_steps_per_iteration (line 338) | def mini_batch_steps_per_iteration(self):
  class KMeansCosineDistanceTest (line 343) | class KMeansCosineDistanceTest(KMeansTestBase):
    method setUp (line 345) | def setUp(self):
    method test_fit (line 370) | def test_fit(self):
    method test_transform (line 378) | def test_transform(self):
    method test_predict (line 387) | def test_predict(self):
    method test_predict_kmeans_plus_plus (line 407) | def test_predict_kmeans_plus_plus(self):
  class MiniBatchKMeansCosineTest (line 449) | class MiniBatchKMeansCosineTest(KMeansCosineDistanceTest):
    method batch_size (line 452) | def batch_size(self):
    method use_mini_batch (line 456) | def use_mini_batch(self):
  class FullBatchAsyncKMeansCosineTest (line 461) | class FullBatchAsyncKMeansCosineTest(KMeansCosineDistanceTest):
    method batch_size (line 464) | def batch_size(self):
    method use_mini_batch (line 468) | def use_mini_batch(self):
    method mini_batch_steps_per_iteration (line 472) | def mini_batch_steps_per_iteration(self):
  class KMeansBenchmark (line 476) | class KMeansBenchmark(benchmark.Benchmark):
    method SetUp (line 479) | def SetUp(self,
    method _report (line 494) | def _report(self, num_iters, start, end, scores):
    method _fit (line 504) | def _fit(self, num_iters=10):
    method benchmark_01_2dim_5center_500point (line 507) | def benchmark_01_2dim_5center_500point(self):
    method benchmark_02_20dim_20center_10kpoint (line 511) | def benchmark_02_20dim_20center_10kpoint(self):
    method benchmark_03_100dim_50center_50kpoint (line 515) | def benchmark_03_100dim_50center_50kpoint(self):
    method benchmark_03_100dim_50center_50kpoint_unseparated (line 519) | def benchmark_03_100dim_50center_50kpoint_unseparated(self):
    method benchmark_04_100dim_500center_500kpoint (line 527) | def benchmark_04_100dim_500center_500kpoint(self):
    method benchmark_05_100dim_500center_500kpoint_unseparated (line 531) | def benchmark_05_100dim_500center_500kpoint_unseparated(self):
  class TensorflowKMeansBenchmark (line 540) | class TensorflowKMeansBenchmark(KMeansBenchmark):
    method _fit (line 542) | def _fit(self, num_iters=10):
  class SklearnKMeansBenchmark (line 562) | class SklearnKMeansBenchmark(KMeansBenchmark):
    method _fit (line 564) | def _fit(self, num_iters=10):
  class KMeansTestQueues (line 582) | class KMeansTestQueues(tf.test.TestCase):
    method input_fn (line 584) | def input_fn(self):
    method test_queues (line 600) | def test_queues(self):

FILE: tensorflow_estimator/python/estimator/canned/linear.py
  class LinearSDCA (line 47) | class LinearSDCA(object):
    method __init__ (line 107) | def __init__(self,
    method _prune_and_unique_sparse_ids (line 135) | def _prune_and_unique_sparse_ids(self, id_weight_pair):
    method get_train_step (line 177) | def get_train_step(self, state_manager, weight_column_name, loss_type,
  function _get_default_optimizer_v2 (line 242) | def _get_default_optimizer_v2(feature_columns):
  function _get_default_optimizer (line 247) | def _get_default_optimizer(feature_columns):
  function _get_expanded_variable_list (line 252) | def _get_expanded_variable_list(var_list):
  function _compute_fraction_of_zero (line 274) | def _compute_fraction_of_zero(variables):
  function linear_logit_fn_builder_v2 (line 311) | def linear_logit_fn_builder_v2(units, feature_columns, sparse_combiner='...
  function linear_logit_fn_builder (line 379) | def linear_logit_fn_builder(units, feature_columns, sparse_combiner='sum'):
  function _sdca_model_fn (line 451) | def _sdca_model_fn(features, labels, mode, head, feature_columns, optimi...
  class _SDCAUpdateWeightsHook (line 543) | class _SDCAUpdateWeightsHook(tf.compat.v1.train.SessionRunHook):
    method __init__ (line 546) | def __init__(self, sdca_model, train_op):
    method begin (line 550) | def begin(self):
    method before_run (line 557) | def before_run(self, run_context):
  function _linear_model_fn_builder_v2 (line 562) | def _linear_model_fn_builder_v2(units,
  function _linear_model_fn_v2 (line 622) | def _linear_model_fn_v2(features,
  function _linear_model_fn (line 687) | def _linear_model_fn(features,
  function _validate_linear_sdca_optimizer_for_linear_classifier (line 756) | def _validate_linear_sdca_optimizer_for_linear_classifier(
  class LinearClassifierV2 (line 771) | class LinearClassifierV2(estimator.EstimatorV2):
    method __init__ (line 858) | def __init__(self,
  class LinearClassifier (line 953) | class LinearClassifier(estimator.Estimator):
    method __init__ (line 956) | def __init__(self,
  class LinearEstimatorV2 (line 999) | class LinearEstimatorV2(estimator.EstimatorV2):
    method __init__ (line 1078) | def __init__(self,
  class LinearEstimator (line 1131) | class LinearEstimator(estimator.Estimator):
    method __init__ (line 1134) | def __init__(self,
  function _validate_linear_sdca_optimizer_for_linear_regressor (line 1189) | def _validate_linear_sdca_optimizer_for_linear_regressor(
  class LinearRegressorV2 (line 1205) | class LinearRegressorV2(estimator.EstimatorV2):
    method __init__ (line 1292) | def __init__(self,
  class LinearRegressor (line 1371) | class LinearRegressor(estimator.Estimator):
    method __init__ (line 1374) | def __init__(self,
  class _LinearModelLayer (line 1417) | class _LinearModelLayer(tf_keras.layers.Layer):
    method __init__ (line 1420) | def __init__(self,
    method build (line 1443) | def build(self, _):
    method call (line 1483) | def call(self, features):
    method get_config (line 1513) | def get_config(self):
    method from_config (line 1529) | def from_config(cls, config, custom_objects=None):
  class LinearModel (line 1540) | class LinearModel(tf_keras.Model):
    method __init__ (line 1579) | def __init__(self,
    method call (line 1653) | def call(self, features):
    method bias (line 1672) | def bias(self):

FILE: tensorflow_estimator/python/estimator/canned/linear_estimator_test.py
  function _linear_estimator_fn (line 37) | def _linear_estimator_fn(weight_column=None, label_dimension=1, **kwargs):
  function _linear_estimator_classifier_fn (line 48) | def _linear_estimator_classifier_fn(n_classes=3, **kwargs):
  class LinearEstimatorEvaluateTest (line 53) | class LinearEstimatorEvaluateTest(
    method __init__ (line 56) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearEstimatorPredictTest (line 62) | class LinearEstimatorPredictTest(
    method __init__ (line 65) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearEstimatorTrainTest (line 71) | class LinearEstimatorTrainTest(
    method __init__ (line 74) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearEstimatorWarmStartingTest (line 80) | class LinearEstimatorWarmStartingTest(
    method __init__ (line 83) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearEstimatorIntegrationTest (line 91) | class LinearEstimatorIntegrationTest(tf.test.TestCase):
    method setUp (line 93) | def setUp(self):
    method tearDown (line 96) | def tearDown(self):
    method _test_complete_flow (line 101) | def _test_complete_flow(self,
    method _create_input_fn (line 143) | def _create_input_fn(self, label_dimension, batch_size):
    method test_numpy_input_fn (line 161) | def test_numpy_input_fn(self):
    method test_numpy_input_fn_with_optimizer_instance (line 175) | def test_numpy_input_fn_with_optimizer_instance(self):

FILE: tensorflow_estimator/python/estimator/canned/linear_model_test.py
  function _initialized_session (line 33) | def _initialized_session(config=None):
  function get_linear_model_bias (line 40) | def get_linear_model_bias(name='linear_model'):
  function get_linear_model_column_var (line 45) | def get_linear_model_column_var(column, name='linear_model'):
  class BaseFeatureColumnForTests (line 50) | class BaseFeatureColumnForTests(tf.compat.v2.__internal__.feature_column...
    method parents (line 58) | def parents(self):
    method from_config (line 62) | def from_config(cls, config, custom_objects=None, columns_by_name=None):
    method get_config (line 65) | def get_config(self):
  class SortableFeatureColumnTest (line 69) | class SortableFeatureColumnTest(tf.test.TestCase):
    method test_linear_model (line 72) | def test_linear_model(self):
    method test_linear_model_sanitizes_scope_names (line 87) | def test_linear_model_sanitizes_scope_names(self):
  class BucketizedColumnTest (line 102) | class BucketizedColumnTest(tf.test.TestCase):
    method test_linear_model_one_input_value (line 104) | def test_linear_model_one_input_value(self):
    method test_linear_model_two_input_values (line 132) | def test_linear_model_two_input_values(self):
  class HashedCategoricalColumnTest (line 162) | class HashedCategoricalColumnTest(tf.test.TestCase):
    method test_linear_model (line 165) | def test_linear_model(self):
  class CrossedColumnTest (line 191) | class CrossedColumnTest(tf.test.TestCase):
    method test_linear_model (line 194) | def test_linear_model(self):
    method test_linear_model_with_weights (line 225) | def test_linear_model_with_weights(self):
  class LinearModelTest (line 289) | class LinearModelTest(tf.test.TestCase):
    method test_raises_if_empty_feature_columns (line 291) | def test_raises_if_empty_feature_columns(self):
    method test_should_be_feature_column (line 296) | def test_should_be_feature_column(self):
    method test_should_be_dense_or_categorical_column (line 300) | def test_should_be_dense_or_categorical_column(self):
    method test_does_not_support_dict_columns (line 323) | def test_does_not_support_dict_columns(self):
    method test_raises_if_duplicate_name (line 328) | def test_raises_if_duplicate_name(self):
    method test_not_dict_input_features (line 335) | def test_not_dict_input_features(self):
    method test_dense_bias (line 343) | def test_dense_bias(self):
    method test_sparse_bias (line 356) | def test_sparse_bias(self):
    method test_dense_and_sparse_bias (line 375) | def test_dense_and_sparse_bias(self):
    method test_dense_and_sparse_column (line 393) | def test_dense_and_sparse_column(self):
    method test_dense_multi_output (line 449) | def test_dense_multi_output(self):
    method test_sparse_multi_output (line 464) | def test_sparse_multi_output(self):
    method test_dense_multi_dimension (line 486) | def test_dense_multi_dimension(self):
    method test_sparse_multi_rank (line 498) | def test_sparse_multi_rank(self):
    method test_sparse_combiner (line 520) | def test_sparse_combiner(self):
    method test_sparse_combiner_sqrtn (line 536) | def test_sparse_combiner_sqrtn(self):
    method test_sparse_combiner_with_negative_weights (line 552) | def test_sparse_combiner_with_negative_weights(self):
    method test_dense_multi_dimension_multi_output (line 573) | def test_dense_multi_dimension_multi_output(self):
    method test_raises_if_shape_mismatch (line 588) | def test_raises_if_shape_mismatch(self):
    method test_dense_reshaping (line 598) | def test_dense_reshaping(self):
    method test_dense_multi_column (line 612) | def test_dense_multi_column(self):
    method test_dense_trainable_default (line 630) | def test_dense_trainable_default(self):
    method test_sparse_trainable_default (line 641) | def test_sparse_trainable_default(self):
    method test_dense_trainable_false (line 654) | def test_dense_trainable_false(self):
    method test_sparse_trainable_false (line 663) | def test_sparse_trainable_false(self):
    method test_column_order (line 674) | def test_column_order(self):
    method test_variable_names (line 710) | def test_variable_names(self):
    method test_fit_and_predict (line 740) | def test_fit_and_predict(self):
    method test_static_batch_size_mismatch (line 756) | def test_static_batch_size_mismatch(self):
    method test_subset_of_static_batch_size_mismatch (line 770) | def test_subset_of_static_batch_size_mismatch(self):
    method test_runtime_batch_size_mismatch (line 786) | def test_runtime_batch_size_mismatch(self):
    method test_runtime_batch_size_matches (line 802) | def test_runtime_batch_size_matches(self):
    method test_with_1d_sparse_tensor (line 821) | def test_with_1d_sparse_tensor(self):
    method test_with_1d_unknown_shape_sparse_tensor (line 861) | def test_with_1d_unknown_shape_sparse_tensor(self):
    method test_with_rank_0_feature (line 906) | def test_with_rank_0_feature(self):
    method test_multiple_linear_models (line 929) | def test_multiple_linear_models(self):
  class VocabularyFileCategoricalColumnTest (line 951) | class VocabularyFileCategoricalColumnTest(tf.test.TestCase):
    method setUp (line 953) | def setUp(self):
    method DISABLED_test_linear_model (line 968) | def DISABLED_test_linear_model(self):
  class VocabularyListCategoricalColumnTest (line 998) | class VocabularyListCategoricalColumnTest(tf.test.TestCase):
    method test_linear_model (line 1001) | def test_linear_model(self):
  class IdentityCategoricalColumnTest (line 1030) | class IdentityCategoricalColumnTest(tf.test.TestCase):
    method test_linear_model (line 1033) | def test_linear_model(self):
  class IndicatorColumnTest (line 1059) | class IndicatorColumnTest(tf.test.TestCase):
    method test_linear_model (line 1062) | def test_linear_model(self):
  class EmbeddingColumnTest (line 1086) | class EmbeddingColumnTest(tf.test.TestCase, parameterized.TestCase):
    method test_linear_model (line 1089) | def test_linear_model(self):
  class SharedEmbeddingColumnTest (line 1171) | class SharedEmbeddingColumnTest(tf.test.TestCase, parameterized.TestCase):
    method test_linear_model (line 1174) | def test_linear_model(self):
  class WeightedCategoricalColumnTest (line 1272) | class WeightedCategoricalColumnTest(tf.test.TestCase):
    method test_linear_model (line 1275) | def test_linear_model(self):
    method test_linear_model_mismatched_shape (line 1308) | def test_linear_model_mismatched_shape(self):
    method test_linear_model_mismatched_dense_values (line 1330) | def test_linear_model_mismatched_dense_values(self):
    method test_linear_model_mismatched_dense_shape (line 1354) | def test_linear_model_mismatched_dense_shape(self):
  class LinearModelLayerSerializationTest (line 1385) | class LinearModelLayerSerializationTest(tf.test.TestCase, parameterized....
    method test_get_config (line 1390) | def test_get_config(self, units, sparse_combiner, trainable, name):
    method test_from_config (line 1411) | def test_from_config(self, units, sparse_combiner, trainable, name):

FILE: tensorflow_estimator/python/estimator/canned/linear_optimizer/python/sdca_test.py
  class SDCAClassifierTest (line 26) | class SDCAClassifierTest(tf.test.TestCase):
    method testRealValuedFeatures (line 28) | def testRealValuedFeatures(self):
    method testRealValuedFeatureWithHigherDimension (line 50) | def testRealValuedFeatureWithHigherDimension(self):
    method testBucketizedFeatures (line 70) | def testBucketizedFeatures(self):
    method testSparseFeatures (line 95) | def testSparseFeatures(self):
    method testWeightedSparseFeatures (line 121) | def testWeightedSparseFeatures(self):
    method testWeightedSparseFeaturesOOVWithNoOOVBuckets (line 152) | def testWeightedSparseFeaturesOOVWithNoOOVBuckets(self):
    method testCrossedFeatures (line 184) | def testCrossedFeatures(self):
    method testMixedFeatures (line 213) | def testMixedFeatures(self):
    method testPartitionedVariables (line 251) | def testPartitionedVariables(self):
  class SDCARegressorTest (line 293) | class SDCARegressorTest(tf.test.TestCase):
    method testRealValuedLinearFeatures (line 295) | def testRealValuedLinearFeatures(self):
    method testMixedFeaturesArbitraryWeights (line 326) | def testMixedFeaturesArbitraryWeights(self):
    method testPartitionedVariables (line 364) | def testPartitionedVariables(self):
    method testSparseFeaturesWithL1Reg (line 404) | def testSparseFeaturesWithL1Reg(self):
    method testBiasOnly (line 478) | def testBiasOnly(self):
    method testBiasAndOtherColumns (line 509) | def testBiasAndOtherColumns(self):
    method testBiasAndOtherColumnsFabricatedCentered (line 571) | def testBiasAndOtherColumnsFabricatedCentered(self):
    method testUnknownBatchSize (line 625) | def testUnknownBatchSize(self):

FILE: tensorflow_estimator/python/estimator/canned/linear_optimizer/python/utils/sdca_ops.py
  class _SparseFeatureColumn (line 34) | class _SparseFeatureColumn(object):
    method __init__ (line 81) | def __init__(self, example_indices, feature_indices, feature_values):
    method example_indices (line 108) | def example_indices(self):
    method feature_indices (line 117) | def feature_indices(self):
    method feature_values (line 126) | def feature_values(self):
  class _SDCAModel (line 135) | class _SDCAModel(object):
    method __init__ (line 197) | def __init__(self, examples, variables, options):
    method _symmetric_l1_regularization (line 253) | def _symmetric_l1_regularization(self):
    method _symmetric_l2_regularization (line 256) | def _symmetric_l2_regularization(self):
    method _num_loss_partitions (line 259) | def _num_loss_partitions(self):
    method _adaptive (line 263) | def _adaptive(self):
    method _num_table_shards (line 267) | def _num_table_shards(self):
    method _create_slots (line 273) | def _create_slots(self):
    method _assert_specified (line 312) | def _assert_specified(self, items, check_in):
    method _assert_list (line 317) | def _assert_list(self, items, check_in):
    method _var_to_list (line 322) | def _var_to_list(self, var):
    method _l1_loss (line 328) | def _l1_loss(self):
    method _l2_loss (line 343) | def _l2_loss(self):
    method _convert_n_to_tensor (line 358) | def _convert_n_to_tensor(self, input_list, as_ref=False):
    method _get_first_dimension_size_statically (line 374) | def _get_first_dimension_size_statically(self, w, num_partitions):
    method _linear_predictions (line 381) | def _linear_predictions(self, examples):
    method predictions (line 412) | def predictions(self, examples):
    method _get_partitioned_update_ops (line 443) | def _get_partitioned_update_ops(self, v_num, num_partitions_by_var,
    method minimize (line 459) | def minimize(self, global_step=None, name=None):
    method update_weights (line 639) | def update_weights(self, train_op):
    method approximate_duality_gap (line 680) | def approximate_duality_gap(self):
    method unregularized_loss (line 710) | def unregularized_loss(self, examples):
    method regularized_loss (line 771) | def regularized_loss(self, examples):

FILE: tensorflow_estimator/python/estimator/canned/linear_optimizer/python/utils/sdca_ops_test.py
  function make_example_proto (line 38) | def make_example_proto(feature_dict, target, value=1.0):
  function make_example_dict (line 52) | def make_example_dict(example_protos, example_weights):
  function make_random_examples_and_variables_dicts (line 98) | def make_random_examples_and_variables_dicts(num_examples, dim, num_non_...
  function make_variable_dict (line 124) | def make_variable_dict(max_age, max_gender, num_shards=None, partitioned...
  function make_dense_examples_and_variables_dicts (line 145) | def make_dense_examples_and_variables_dicts(dense_features_values, weights,
  function get_binary_predictions_for_logistic (line 190) | def get_binary_predictions_for_logistic(predictions, cutoff=0.5):
  function get_binary_predictions_for_hinge (line 197) | def get_binary_predictions_for_hinge(predictions):
  class _SDCAModelTest (line 206) | class _SDCAModelTest(TensorFlowTestCase):
    method _single_threaded_test_session (line 209) | def _single_threaded_test_session(self):
  class SdcaWithLogisticLossTest (line 217) | class SdcaWithLogisticLossTest(_SDCAModelTest):
    method testSimple (line 220) | def testSimple(self):
    method testPartitionedPrimals (line 268) | def testPartitionedPrimals(self):
    method testSomePartitionedPrimals (line 316) | def testSomePartitionedPrimals(self):
    method testSparseRandom (line 377) | def testSparseRandom(self):
    method testSparseDuplicate (line 401) | def testSparseDuplicate(self):
    method testDistributedSimple (line 428) | def testDistributedSimple(self):
    method testSimpleNoL2 (line 504) | def testSimpleNoL2(self):
    method testSomeUnweightedExamples (line 529) | def testSomeUnweightedExamples(self):
    method testFractionalExampleLabel (line 583) | def testFractionalExampleLabel(self):
    method testImbalanced (line 612) | def testImbalanced(self):
    method testImbalancedWithExampleWeights (line 661) | def testImbalancedWithExampleWeights(self):
    method testInstancesOfOneClassOnly (line 701) | def testInstancesOfOneClassOnly(self):
    method testOutOfRangeSparseFeatures (line 740) | def testOutOfRangeSparseFeatures(self):
    method testOutOfRangeDenseFeatures (line 767) | def testOutOfRangeDenseFeatures(self):
    method testMissingFeature (line 789) | def testMissingFeature(self):
  class SdcaWithLinearLossTest (line 832) | class SdcaWithLinearLossTest(_SDCAModelTest):
    method testSimple (line 835) | def testSimple(self):
    method testL2Regularization (line 874) | def testL2Regularization(self):
    method testL1Regularization (line 922) | def testL1Regularization(self):
    method testFeatureValues (line 960) | def testFeatureValues(self):
    method testDenseFeaturesWithDefaultWeights (line 1007) | def testDenseFeaturesWithDefaultWeights(self):
    method testDenseFeaturesWithArbitraryWeights (line 1037) | def testDenseFeaturesWithArbitraryWeights(self):
  class SdcaWithHingeLossTest (line 1072) | class SdcaWithHingeLossTest(_SDCAModelTest):
    method testSimple (line 1075) | def testSimple(self):
    method testDenseFeaturesPerfectlySeparable (line 1123) | def testDenseFeaturesPerfectlySeparable(self):
    method testDenseFeaturesSeparableWithinMargins (line 1155) | def testDenseFeaturesSeparableWithinMargins(self):
    method testDenseFeaturesWeightedExamples (line 1186) | def testDenseFeaturesWeightedExamples(self):
  class SdcaWithSmoothHingeLossTest (line 1222) | class SdcaWithSmoothHingeLossTest(_SDCAModelTest):
    method testSimple (line 1225) | def testSimple(self):
  class SdcaWithPoissonLossTest (line 1275) | class SdcaWithPoissonLossTest(_SDCAModelTest):
    method testSimple (line 1278) | def testSimple(self):
  class SdcaFprintTest (line 1327) | class SdcaFprintTest(_SDCAModelTest):
    method testFprint (line 1337) | def testFprint(self):
  class _SparseFeatureColumnTest (line 1347) | class _SparseFeatureColumnTest(TensorFlowTestCase):
    method testBasic (line 1350) | def testBasic(self):

FILE: tensorflow_estimator/python/estimator/canned/linear_optimizer/python/utils/sharded_mutable_dense_hashtable.py
  class _MutableDenseHashTable (line 33) | class _MutableDenseHashTable(lookup_ops.LookupInterface):
    method __init__ (line 37) | def __init__(self,
    method _create_resource (line 104) | def _create_resource(self):
    method name (line 125) | def name(self):
    method size (line 128) | def size(self, name=None):
    method lookup (line 143) | def lookup(self, keys, name=None):
    method insert (line 169) | def insert(self, keys, values, name=None):
    method export (line 196) | def export(self, name=None):
    method _serialize_to_tensors (line 214) | def _serialize_to_tensors(self):
    method _restore_from_tensors (line 218) | def _restore_from_tensors(self, restored_tensors):
    class _Saveable (line 224) | class _Saveable(BaseSaverBuilder.SaveableObject):
      method __init__ (line 227) | def __init__(self, table, name):
      method restore (line 236) | def restore(self, restored_tensors, restored_shapes):
  class _ShardedMutableDenseHashTable (line 247) | class _ShardedMutableDenseHashTable(object):
    method __init__ (line 258) | def __init__(self,
    method name (line 289) | def name(self):
    method _num_shards (line 293) | def _num_shards(self):
    method table_shards (line 297) | def table_shards(self):
    method size (line 300) | def size(self, name=None):
    method _shard_indices (line 305) | def _shard_indices(self, keys):
    method _check_keys (line 314) | def _check_keys(self, keys):
    method lookup (line 319) | def lookup(self, keys, name=None):
    method insert (line 342) | def insert(self, keys, values, name=None):
    method export_sharded (line 359) | def export_sharded(self, name=None):

FILE: tensorflow_estimator/python/estimator/canned/linear_optimizer/python/utils/sharded_mutable_dense_hashtable_test.py
  class _ShardedMutableDenseHashTableTest (line 26) | class _ShardedMutableDenseHashTableTest(tf.test.TestCase):
    method testShardedMutableHashTable (line 29) | def testShardedMutableHashTable(self):
    method testShardedMutableHashTableVectors (line 54) | def testShardedMutableHashTableVectors(self):
    method testExportSharded (line 82) | def testExportSharded(self):

FILE: tensorflow_estimator/python/estimator/canned/linear_test.py
  function _linear_regressor_fn (line 29) | def _linear_regressor_fn(*args, **kwargs):
  function _linear_classifier_fn (line 33) | def _linear_classifier_fn(*args, **kwargs):
  class LinearRegressorEvaluationV2Test (line 40) | class LinearRegressorEvaluationV2Test(
    method __init__ (line 43) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearRegressorPredictV2Test (line 49) | class LinearRegressorPredictV2Test(
    method __init__ (line 52) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearRegressorIntegrationV2Test (line 58) | class LinearRegressorIntegrationV2Test(
    method __init__ (line 61) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearRegressorTrainingV2Test (line 67) | class LinearRegressorTrainingV2Test(
    method __init__ (line 70) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearClassifierTrainingV2Test (line 79) | class LinearClassifierTrainingV2Test(
    method __init__ (line 82) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearClassifierEvaluationV2Test (line 90) | class LinearClassifierEvaluationV2Test(
    method __init__ (line 93) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearClassifierPredictV2Test (line 101) | class LinearClassifierPredictV2Test(
    method __init__ (line 104) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearClassifierIntegrationV2Test (line 112) | class LinearClassifierIntegrationV2Test(
    method __init__ (line 115) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearLogitFnV2Test (line 126) | class LinearLogitFnV2Test(linear_testing_utils.BaseLinearLogitFnTest,
    method __init__ (line 129) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearWarmStartingV2Test (line 138) | class LinearWarmStartingV2Test(linear_testing_utils.BaseLinearWarmStarti...
    method __init__ (line 141) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class ComputeFractionOfZeroTest (line 150) | class ComputeFractionOfZeroTest(tf.test.TestCase):
    method _assertSparsity (line 152) | def _assertSparsity(self, expected_sparsity, tensor):
    method test_small_float32 (line 156) | def test_small_float32(self):
    method test_small_int32 (line 162) | def test_small_int32(self):
    method test_small_float64 (line 166) | def test_small_float64(self):
    method test_small_int64 (line 170) | def test_small_int64(self):
    method test_nested (line 174) | def test_nested(self):
    method test_none (line 179) | def test_none(self):
    method test_empty (line 183) | def test_empty(self):
    method test_multiple_empty (line 189) | def test_multiple_empty(self):
    method test_some_empty (line 198) | def test_some_empty(self):
    method test_mixed_types (line 206) | def test_mixed_types(self):
    method test_2_27_zeros__using_512_MiB_of_ram (line 213) | def test_2_27_zeros__using_512_MiB_of_ram(self):
    method test_2_27_ones__using_512_MiB_of_ram (line 217) | def test_2_27_ones__using_512_MiB_of_ram(self):

FILE: tensorflow_estimator/python/estimator/canned/linear_testing_utils.py
  function assert_close (line 62) | def assert_close(expected, actual, rtol=1e-04, name='assert_close'):
  function save_variables_to_ckpt (line 77) | def save_variables_to_ckpt(model_dir):
  function queue_parsed_features (line 84) | def queue_parsed_features(feature_map):
  function sorted_key_dict (line 99) | def sorted_key_dict(unsorted_dict):
  function sigmoid (line 103) | def sigmoid(x):
  function mock_optimizer (line 107) | def mock_optimizer(testcase, expected_loss=None):
  class BaseLinearRegressorEvaluationTest (line 142) | class BaseLinearRegressorEvaluationTest(object):
    method __init__ (line 144) | def __init__(self, linear_regressor_fn, fc_lib=feature_column_v2):
    method setUp (line 148) | def setUp(self):
    method tearDown (line 151) | def tearDown(self):
    method test_evaluation_for_simple_data (line 156) | def test_evaluation_for_simple_data(self):
    method test_evaluation_batch (line 182) | def test_evaluation_batch(self):
    method test_evaluation_weights (line 212) | def test_evaluation_weights(self):
    method test_evaluation_for_multi_dimensions (line 246) | def test_evaluation_for_multi_dimensions(self):
    method test_evaluation_for_multiple_feature_columns (line 282) | def test_evaluation_for_multiple_feature_columns(self):
    method test_evaluation_for_multiple_feature_columns_mix (line 320) | def test_evaluation_for_multiple_feature_columns_mix(self):
  class BaseLinearRegressorPredictTest (line 360) | class BaseLinearRegressorPredictTest(object):
    method __init__ (line 362) | def __init__(self, linear_regressor_fn, fc_lib=feature_column_v2):
    method setUp (line 366) | def setUp(self):
    method tearDown (line 369) | def tearDown(self):
    method test_1d (line 374) | def test_1d(self):
    method testMultiDim (line 397) | def testMultiDim(self):
    method testTwoFeatureColumns (line 430) | def testTwoFeatureColumns(self):
    method testTwoFeatureColumnsMix (line 458) | def testTwoFeatureColumnsMix(self):
    method testSparseCombiner (line 483) | def testSparseCombiner(self):
  class BaseLinearRegressorIntegrationTest (line 546) | class BaseLinearRegressorIntegrationTest(object):
    method __init__ (line 548) | def __init__(self, linear_regressor_fn, fc_lib=feature_column_v2):
    method setUp (line 552) | def setUp(self):
    method tearDown (line 555) | def tearDown(self):
    method _test_complete_flow (line 560) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method test_numpy_input_fn (line 592) | def test_numpy_input_fn(self):
    method test_pandas_input_fn (line 628) | def test_pandas_input_fn(self):
    method test_input_fn_from_parse_example (line 657) | def test_input_fn_from_parse_example(self):
  class BaseLinearRegressorTrainingTest (line 718) | class BaseLinearRegressorTrainingTest(object):
    method __init__ (line 720) | def __init__(self, linear_regressor_fn, fc_lib=feature_column_v2):
    method setUp (line 724) | def setUp(self):
    method tearDown (line 727) | def tearDown(self):
    method _assert_checkpoint (line 732) | def _assert_checkpoint(self,
    method testFromScratchWithDefaultOptimizer (line 757) | def testFromScratchWithDefaultOptimizer(self):
    method testTrainWithOneDimLabel (line 773) | def testTrainWithOneDimLabel(self):
    method testTrainWithOneDimWeight (line 793) | def testTrainWithOneDimWeight(self):
    method testFromScratch (line 818) | def testFromScratch(self):
    method testFromCheckpoint (line 843) | def testFromCheckpoint(self):
    method testFromCheckpointMultiBatch (line 879) | def testFromCheckpointMultiBatch(self):
  class BaseLinearClassifierTrainingTest (line 920) | class BaseLinearClassifierTrainingTest(object):
    method __init__ (line 922) | def __init__(self, linear_classifier_fn, fc_lib=feature_column_v2):
    method setUp (line 926) | def setUp(self):
    method tearDown (line 929) | def tearDown(self):
    method _assert_checkpoint (line 933) | def _assert_checkpoint(self,
    method _testFromScratchWithDefaultOptimizer (line 962) | def _testFromScratchWithDefaultOptimizer(self, n_classes):
    method testBinaryClassesFromScratchWithDefaultOptimizer (line 978) | def testBinaryClassesFromScratchWithDefaultOptimizer(self):
    method testMultiClassesFromScratchWithDefaultOptimizer (line 981) | def testMultiClassesFromScratchWithDefaultOptimizer(self):
    method _testTrainWithTwoDimsLabel (line 984) | def _testTrainWithTwoDimsLabel(self, n_classes):
    method testBinaryClassesTrainWithTwoDimsLabel (line 1005) | def testBinaryClassesTrainWithTwoDimsLabel(self):
    method testMultiClassesTrainWithTwoDimsLabel (line 1008) | def testMultiClassesTrainWithTwoDimsLabel(self):
    method _testTrainWithOneDimLabel (line 1011) | def _testTrainWithOneDimLabel(self, n_classes):
    method testBinaryClassesTrainWithOneDimLabel (line 1030) | def testBinaryClassesTrainWithOneDimLabel(self):
    method testMultiClassesTrainWithOneDimLabel (line 1033) | def testMultiClassesTrainWithOneDimLabel(self):
    method _testTrainWithTwoDimsWeight (line 1036) | def _testTrainWithTwoDimsWeight(self, n_classes):
    method testBinaryClassesTrainWithTwoDimsWeight (line 1061) | def testBinaryClassesTrainWithTwoDimsWeight(self):
    method testMultiClassesTrainWithTwoDimsWeight (line 1064) | def testMultiClassesTrainWithTwoDimsWeight(self):
    method _testTrainWithOneDimWeight (line 1067) | def _testTrainWithOneDimWeight(self, n_classes):
    method testBinaryClassesTrainWithOneDimWeight (line 1090) | def testBinaryClassesTrainWithOneDimWeight(self):
    method testMultiClassesTrainWithOneDimWeight (line 1093) | def testMultiClassesTrainWithOneDimWeight(self):
    method _testFromScratch (line 1096) | def _testFromScratch(self, n_classes):
    method testBinaryClassesFromScratch (line 1132) | def testBinaryClassesFromScratch(self):
    method testMultiClassesFromScratch (line 1135) | def testMultiClassesFromScratch(self):
    method _testFromCheckpoint (line 1138) | def _testFromCheckpoint(self, n_classes):
    method testBinaryClassesFromCheckpoint (line 1197) | def testBinaryClassesFromCheckpoint(self):
    method testMultiClassesFromCheckpoint (line 1200) | def testMultiClassesFromCheckpoint(self):
    method _testFromCheckpointFloatLabels (line 1203) | def _testFromCheckpointFloatLabels(self, n_classes):
    method testBinaryClassesFromCheckpointFloatLabels (line 1242) | def testBinaryClassesFromCheckpointFloatLabels(self):
    method testMultiClassesFromCheckpointFloatLabels (line 1245) | def testMultiClassesFromCheckpointFloatLabels(self):
    method _testFromCheckpointMultiBatch (line 1248) | def _testFromCheckpointMultiBatch(self, n_classes):
    method testBinaryClassesFromCheckpointMultiBatch (line 1312) | def testBinaryClassesFromCheckpointMultiBatch(self):
    method testMultiClassesFromCheckpointMultiBatch (line 1315) | def testMultiClassesFromCheckpointMultiBatch(self):
  class BaseLinearClassifierEvaluationTest (line 1319) | class BaseLinearClassifierEvaluationTest(object):
    method __init__ (line 1321) | def __init__(self, linear_classifier_fn, fc_lib=feature_column_v2):
    method setUp (line 1325) | def setUp(self):
    method tearDown (line 1328) | def tearDown(self):
    method _test_evaluation_for_simple_data (line 1332) | def _test_evaluation_for_simple_data(self, n_classes):
    method test_binary_classes_evaluation_for_simple_data (line 1394) | def test_binary_classes_evaluation_for_simple_data(self):
    method test_multi_classes_evaluation_for_simple_data (line 1397) | def test_multi_classes_evaluation_for_simple_data(self):
    method _test_evaluation_batch (line 1400) | def _test_evaluation_batch(self, n_classes):
    method test_binary_classes_evaluation_batch (line 1472) | def test_binary_classes_evaluation_batch(self):
    method test_multi_classes_evaluation_batch (line 1475) | def test_multi_classes_evaluation_batch(self):
    method _test_evaluation_weights (line 1478) | def _test_evaluation_weights(self, n_classes):
    method test_binary_classes_evaluation_weights (line 1563) | def test_binary_classes_evaluation_weights(self):
    method test_multi_classes_evaluation_weights (line 1566) | def test_multi_classes_evaluation_weights(self):
  class BaseLinearClassifierPredictTest (line 1570) | class BaseLinearClassifierPredictTest(object):
    method __init__ (line 1572) | def __init__(self, linear_classifier_fn, fc_lib=feature_column_v2):
    method setUp (line 1576) | def setUp(self):
    method tearDown (line 1579) | def tearDown(self):
    method _testPredictions (line 1583) | def _testPredictions(self, n_classes, label_vocabulary, label_output_fn):
    method testBinaryClassesWithoutLabelVocabulary (line 1658) | def testBinaryClassesWithoutLabelVocabulary(self):
    method testBinaryClassesWithLabelVocabulary (line 1665) | def testBinaryClassesWithLabelVocabulary(self):
    method testMultiClassesWithoutLabelVocabulary (line 1672) | def testMultiClassesWithoutLabelVocabulary(self):
    method testMultiClassesWithLabelVocabulary (line 1679) | def testMultiClassesWithLabelVocabulary(self):
    method testSparseCombiner (line 1686) | def testSparseCombiner(self):
  class BaseLinearClassifierIntegrationTest (line 1749) | class BaseLinearClassifierIntegrationTest(object):
    method __init__ (line 1751) | def __init__(self, linear_classifier_fn, fc_lib=feature_column_v2):
    method setUp (line 1755) | def setUp(self):
    method tearDown (line 1758) | def tearDown(self):
    method _test_complete_flow (line 1762) | def _test_complete_flow(self, n_classes, train_input_fn, eval_input_fn,
    method _test_numpy_input_fn (line 1794) | def _test_numpy_input_fn(self, n_classes):
    method test_binary_classes_numpy_input_fn (line 1830) | def test_binary_classes_numpy_input_fn(self):
    method test_multi_classes_numpy_input_fn (line 1833) | def test_multi_classes_numpy_input_fn(self):
    method _test_pandas_input_fn (line 1836) | def _test_pandas_input_fn(self, n_classes):
    method test_binary_classes_pandas_input_fn (line 1865) | def test_binary_classes_pandas_input_fn(self):
    method test_multi_classes_pandas_input_fn (line 1868) | def test_multi_classes_pandas_input_fn(self):
    method _test_input_fn_from_parse_example (line 1871) | def _test_input_fn_from_parse_example(self, n_classes):
    method test_binary_classes_input_fn_from_parse_example (line 1930) | def test_binary_classes_input_fn_from_parse_example(self):
    method test_multi_classes_input_fn_from_parse_example (line 1933) | def test_multi_classes_input_fn_from_parse_example(self):
  class BaseLinearLogitFnTest (line 1937) | class BaseLinearLogitFnTest(object):
    method __init__ (line 1939) | def __init__(self, fc_lib=feature_column_v2):
    method test_basic_logit_correctness (line 1942) | def test_basic_logit_correctness(self):
    method test_compute_fraction_of_zero_v2 (line 1963) | def test_compute_fraction_of_zero_v2(self):
  class BaseLinearWarmStartingTest (line 1995) | class BaseLinearWarmStartingTest(object):
    method __init__ (line 1997) | def __init__(self,
    method setUp (line 2005) | def setUp(self):
    method tearDown (line 2020) | def tearDown(self):
    method test_classifier_basic_warm_starting (line 2025) | def test_classifier_basic_warm_starting(self):
    method test_regressor_basic_warm_starting (line 2059) | def test_regressor_basic_warm_starting(self):
    method test_warm_starting_selective_variables (line 2091) | def test_warm_starting_selective_variables(self):
    method test_warm_starting_with_vocab_remapping_and_partitioning (line 2124) | def test_warm_starting_with_vocab_remapping_and_partitioning(self):
    method test_warm_starting_with_naming_change (line 2202) | def test_warm_starting_with_naming_change(self):

FILE: tensorflow_estimator/python/estimator/canned/metric_keys.py
  class MetricKeys (line 24) | class MetricKeys(object):

FILE: tensorflow_estimator/python/estimator/canned/optimizers.py
  function get_optimizer_instance (line 48) | def get_optimizer_instance(opt, learning_rate=None):
  function _optimizer_has_default_learning_rate (line 90) | def _optimizer_has_default_learning_rate(opt):
  function get_optimizer_instance_v2 (line 99) | def get_optimizer_instance_v2(opt, learning_rate=None):

FILE: tensorflow_estimator/python/estimator/canned/optimizers_test.py
  class _TestOptimizer (line 26) | class _TestOptimizer(tf.compat.v1.train.Optimizer):
    method __init__ (line 28) | def __init__(self):
  class GetOptimizerInstance (line 33) | class GetOptimizerInstance(tf.test.TestCase):
    method test_unsupported_name (line 35) | def test_unsupported_name(self):
    method test_supported_name_but_learning_rate_none (line 40) | def test_supported_name_but_learning_rate_none(self):
    method test_keras_optimizer_after_tf_2_11 (line 45) | def test_keras_optimizer_after_tf_2_11(self):
    method test_adagrad (line 61) | def test_adagrad(self):
    method test_adam (line 66) | def test_adam(self):
    method test_ftrl (line 71) | def test_ftrl(self):
    method test_rmsprop (line 76) | def test_rmsprop(self):
    method test_sgd (line 81) | def test_sgd(self):
    method test_object (line 86) | def test_object(self):
    method test_object_invalid (line 90) | def test_object_invalid(self):
    method test_callable (line 95) | def test_callable(self):
    method test_lambda (line 103) | def test_lambda(self):
    method test_callable_returns_invalid (line 107) | def test_callable_returns_invalid(self):

FILE: tensorflow_estimator/python/estimator/canned/optimizers_test_v2.py
  class _TestOptimizerV2 (line 26) | class _TestOptimizerV2(tf_keras.optimizers.legacy.Optimizer):
    method __init__ (line 28) | def __init__(self):
    method get_config (line 31) | def get_config(self):
  class GetOptimizerInstanceV2 (line 35) | class GetOptimizerInstanceV2(tf.test.TestCase):
    method test_unsupported_name (line 38) | def test_unsupported_name(self):
    method test_adagrad_but_no_learning_rate (line 44) | def test_adagrad_but_no_learning_rate(self):
    method test_adam_but_no_learning_rate (line 56) | def test_adam_but_no_learning_rate(self):
    method test_adagrad (line 65) | def test_adagrad(self):
    method test_adam (line 75) | def test_adam(self):
    method test_ftrl (line 84) | def test_ftrl(self):
    method test_rmsprop (line 93) | def test_rmsprop(self):
    method test_sgd (line 103) | def test_sgd(self):
    method test_object (line 112) | def test_object(self):
    method test_object_invalid (line 116) | def test_object_invalid(self):
    method test_callable (line 122) | def test_callable(self):
    method test_lambda (line 130) | def test_lambda(self):
    method test_callable_returns_invalid (line 134) | def test_callable_returns_invalid(self):

FILE: tensorflow_estimator/python/estimator/canned/parsing_utils.py
  function classifier_parse_example_spec_v2 (line 28) | def classifier_parse_example_spec_v2(feature_columns,
  function regressor_parse_example_spec_v2 (line 148) | def regressor_parse_example_spec_v2(feature_columns,
  function _add_label_and_weight_to_parsing_spec (line 264) | def _add_label_and_weight_to_parsing_spec(parsing_spec,
  function classifier_parse_example_spec (line 316) | def classifier_parse_example_spec(feature_columns,
  function regressor_parse_example_spec (line 335) | def regressor_parse_example_spec(

FILE: tensorflow_estimator/python/estimator/canned/parsing_utils_test.py
  class BaseClassifierParseExampleSpec (line 25) | class BaseClassifierParseExampleSpec(object):
    method __init__ (line 28) | def __init__(self, parse_example_fn):
    method test_defaults (line 31) | def test_defaults(self):
    method test_string (line 40) | def test_string(self):
    method test_label_default_value (line 52) | def test_label_default_value(self):
    method test_weight_column_as_string (line 65) | def test_weight_column_as_string(self):
    method test_weight_column_as_numeric_column (line 77) | def test_weight_column_as_numeric_column(self):
    method test_label_key_should_not_be_used_as_feature (line 89) | def test_label_key_should_not_be_used_as_feature(self):
    method test_weight_column_should_not_be_used_as_feature (line 96) | def test_weight_column_should_not_be_used_as_feature(self):
    method test_weight_column_should_be_a_numeric_column (line 104) | def test_weight_column_should_be_a_numeric_column(self):
  class ClassifierParseExampleSpecV2 (line 114) | class ClassifierParseExampleSpecV2(BaseClassifierParseExampleSpec,
    method __init__ (line 117) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
    method test_non_v1_feature_column (line 122) | def test_non_v1_feature_column(self):
  class ClassifierParseExampleSpecV1 (line 133) | class ClassifierParseExampleSpecV1(BaseClassifierParseExampleSpec,
    method __init__ (line 136) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class BaseRegressorParseExampleSpec (line 142) | class BaseRegressorParseExampleSpec(object):
    method __init__ (line 145) | def __init__(self, parse_example_fn):
    method test_defaults (line 148) | def test_defaults(self):
    method test_int64 (line 157) | def test_int64(self):
    method test_label_default_value (line 168) | def test_label_default_value(self):
    method test_label_dimension (line 183) | def test_label_dimension(self):
    method test_weight_column_as_string (line 194) | def test_weight_column_as_string(self):
    method test_weight_column_as_numeric_column (line 206) | def test_weight_column_as_numeric_column(self):
    method test_label_key_should_not_be_used_as_feature (line 218) | def test_label_key_should_not_be_used_as_feature(self):
    method test_weight_column_should_not_be_used_as_feature (line 225) | def test_weight_column_should_not_be_used_as_feature(self):
    method test_weight_column_should_be_a_numeric_column (line 233) | def test_weight_column_should_be_a_numeric_column(self):
  class RegressorParseExampleSpecV2 (line 243) | class RegressorParseExampleSpecV2(BaseRegressorParseExampleSpec,
    method __init__ (line 246) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
    method test_non_v1_feature_column (line 251) | def test_non_v1_feature_column(self):
  class RegressorParseExampleSpecV1 (line 262) | class RegressorParseExampleSpecV1(BaseRegressorParseExampleSpec,
    method __init__ (line 265) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name

FILE: tensorflow_estimator/python/estimator/canned/prediction_keys.py
  class PredictionKeys (line 22) | class PredictionKeys(object):

FILE: tensorflow_estimator/python/estimator/canned/rnn.py
  function _single_rnn_cell (line 63) | def _single_rnn_cell(units, cell_type):
  function _make_rnn_cell_fn (line 81) | def _make_rnn_cell_fn(units, cell_type=_SIMPLE_RNN_KEY):
  class RNNModel (line 105) | class RNNModel(tf_keras.models.Model):
    method __init__ (line 129) | def __init__(self,
    method call (line 180) | def call(self, inputs, training=None):
    method get_config (line 227) | def get_config(self):
    method from_config (line 245) | def from_config(cls, config, custom_objects=None):
  function _get_rnn_estimator_spec (line 274) | def _get_rnn_estimator_spec(features, labels, mode, head, rnn_model, opt...
  function _verify_rnn_cell_input (line 332) | def _verify_rnn_cell_input(rnn_cell_fn, units, cell_type):
  function _make_rnn_layer (line 338) | def _make_rnn_layer(rnn_cell_fn, units, cell_type, return_sequences):
  class RNNEstimator (line 366) | class RNNEstimator(estimator.Estimator):
    method __init__ (line 426) | def __init__(self,
  class RNNClassifier (line 519) | class RNNClassifier(RNNEstimator):
    method __init__ (line 571) | def __init__(self,

FILE: tensorflow_estimator/python/estimator/canned/rnn_test.py
  function _assert_close (line 51) | def _assert_close(expected, actual, rtol=1e-04, name='assert_close'):
  function create_checkpoint (line 66) | def create_checkpoint(kernel, recurrent, bias, dense_kernel, dense_bias,
  function _make_rnn_layer (line 101) | def _make_rnn_layer(rnn_cell_fn=None,
  class RNNLayerFnTest (line 113) | class RNNLayerFnTest(tf.test.TestCase, parameterized.TestCase):
    method testWrongClassProvided (line 116) | def testWrongClassProvided(self):
    method testWrongStringProvided (line 122) | def testWrongStringProvided(self):
    method testDefaultCellProvided (line 130) | def testDefaultCellProvided(self, cell_type):
    method testSpecificLayerTypeProvided (line 139) | def testSpecificLayerTypeProvided(self, cell_type, layer_type):
    method testSpecificLayerTypeArguments (line 144) | def testSpecificLayerTypeArguments(self):
    method testCustomCellProvided (line 157) | def testCustomCellProvided(self, mock_rnn_layer_type):
    method testMultipleCellsProvided (line 167) | def testMultipleCellsProvided(self):
    method testCustomCellFnProvided (line 176) | def testCustomCellFnProvided(self, mock_rnn_layer_type):
  function _mock_logits_layer (line 185) | def _mock_logits_layer(kernel, bias):
  function _default_features_fn (line 205) | def _default_features_fn():
  function _get_mock_head (line 213) | def _get_mock_head():
  class RNNLogitFnTest (line 221) | class RNNLogitFnTest(tf.test.TestCase, parameterized.TestCase):
    method setUp (line 224) | def setUp(self):
    method _mock_logits_layer (line 237) | def _mock_logits_layer(self):
    method _test_logits (line 240) | def _test_logits(self,
    method testOneDimLogits (line 289) | def testOneDimLogits(self, return_sequences, expected_logits, training...
    method testMultiDimLogits (line 332) | def testMultiDimLogits(self, return_sequences, expected_logits):
    method testMultiExampleMultiDim (line 383) | def testMultiExampleMultiDim(self, return_sequences, expected_logits):
    method testMultiExamplesDifferentLength (line 450) | def testMultiExamplesDifferentLength(self, return_sequences, expected_...
    method testMultiExamplesWithContext (line 496) | def testMultiExamplesWithContext(self):
    method testMultiExamplesMultiFeatures (line 539) | def testMultiExamplesMultiFeatures(self):
    method testTrainingMode (line 592) | def testTrainingMode(self, mode, expected_training_mode):
  class RNNModelTest (line 621) | class RNNModelTest(tf.test.TestCase, parameterized.TestCase):
    method setUp (line 624) | def setUp(self):
    method _get_compiled_model (line 643) | def _get_compiled_model(self,
    method testModelWeights (line 669) | def testModelWeights(self):
    method _testModelConfig (line 705) | def _testModelConfig(self, **kwargs):
    method testModelConfig (line 737) | def testModelConfig(self):
    method testModelConfigWithActivation (line 750) | def testModelConfigWithActivation(self):
    method testModelConfigWithContextFeatures (line 755) | def testModelConfigWithContextFeatures(self):
    method DISABLED_testSaveModelWeights (line 762) | def DISABLED_testSaveModelWeights(self):  # See b/129842600.
    method DISABLED_testEvaluationMetrics (line 774) | def DISABLED_testEvaluationMetrics(self):  # See b/129842600.
    method DISABLED_testEvaluationSequential (line 782) | def DISABLED_testEvaluationSequential(self):  # See b/129842600.
    method DISABLED_testPredictions (line 789) | def DISABLED_testPredictions(self):  # See b/129842600.
    method DISABLED_testPredictionsSequential (line 796) | def DISABLED_testPredictionsSequential(self):  # See b/129842600.
    method DISABLED_testTraining (line 808) | def DISABLED_testTraining(self, optimizer):  # See b/129842600.
    method DISABLED_testTrainingSequential (line 819) | def DISABLED_testTrainingSequential(self):  # See b/129842600.
  class RNNEstimatorInitTest (line 828) | class RNNEstimatorInitTest(tf.test.TestCase):
    method setUp (line 830) | def setUp(self):
    method testConflictingRNNCellFn (line 839) | def testConflictingRNNCellFn(self):
    method testNonSequentialHeadProvided (line 856) | def testNonSequentialHeadProvided(self):
    method testWrongOptimizerTypeProvided (line 865) | def testWrongOptimizerTypeProvided(self):
  class RNNClassifierTrainingTest (line 876) | class RNNClassifierTrainingTest(tf.test.TestCase):
    method setUp (line 878) | def setUp(self):
    method _assert_checkpoint (line 889) | def _assert_checkpoint(self, n_classes, input_units, cell_units,
    method _mock_optimizer (line 919) | def _mock_optimizer(self, expected_loss=None):
    method _testFromScratchWithDefaultOptimizer (line 955) | def _testFromScratchWithDefaultOptimizer(self, n_classes):
    method testBinaryClassFromScratchWithDefaultOptimizer (line 983) | def testBinaryClassFromScratchWithDefaultOptimizer(self):
    method testMultiClassFromScratchWithDefaultOptimizer (line 986) | def testMultiClassFromScratchWithDefaultOptimizer(self):
    method testFromScratchWithCustomRNNCellFn (line 989) | def testFromScratchWithCustomRNNCellFn(self):
    method _testExampleWeight (line 1022) | def _testExampleWeight(self, n_classes):
    method testBinaryClassWithExampleWeight (line 1052) | def testBinaryClassWithExampleWeight(self):
    method testMultiClassWithExampleWeight (line 1055) | def testMultiClassWithExampleWeight(self):
    method _testFromCheckpoint (line 1058) | def _testFromCheckpoint(self, input_fn, expected_loss, **kwargs):
    method testBinaryClassFromCheckpoint (line 1081) | def testBinaryClassFromCheckpoint(self):
    method testMultiClassFromCheckpoint (line 1096) | def testMultiClassFromCheckpoint(self):
    method testBinaryClassFromCheckpointSequential (line 1114) | def testBinaryClassFromCheckpointSequential(self):
    method testBinaryClassFromCheckpointSequentialWithWeights (line 1142) | def testBinaryClassFromCheckpointSequentialWithWeights(self):
    method testDefaultGradientClipping (line 1176) | def testDefaultGradientClipping(self):
  function sorted_key_dict (line 1230) | def sorted_key_dict(unsorted_dict):
  class RNNClassifierEvaluationTest (line 1235) | class RNNClassifierEvaluationTest(tf.test.TestCase):
    method setUp (line 1237) | def setUp(self):
    method _testFromCheckpoint (line 1249) | def _testFromCheckpoint(self, input_fn, **kwargs):
    method testBinaryClassEvaluationMetrics (line 1266) | def testBinaryClassEvaluationMetrics(self):
    method testBinaryClassEvaluationMetricsSequential (line 1304) | def testBinaryClassEvaluationMetricsSequential(self):
    method testMultiClassEvaluationMetrics (line 1351) | def testMultiClassEvaluationMetrics(self):
  class RNNClassifierPredictionTest (line 1391) | class RNNClassifierPredictionTest(tf.test.TestCase):
    method setUp (line 1393) | def setUp(self):
    method _testFromCheckpoint (line 1404) | def _testFromCheckpoint(self, input_fn, **kwargs):
    method testBinaryClassPredictions (line 1430) | def testBinaryClassPredictions(self):
    method testMultiClassPredictions (line 1450) | def testMultiClassPredictions(self):
    method testBinaryClassPredictionsSequential (line 1471) | def testBinaryClassPredictionsSequential(self):
  class BaseRNNClassificationIntegrationTest (line 1505) | class BaseRNNClassificationIntegrationTest(object):
    method setUp (line 1507) | def setUp(self):
    method __init__ (line 1514) | def __init__(self, _create_estimator_fn):
    method _test_complete_flow (line 1517) | def _test_complete_flow(self,
    method _testNumpyInputFn (line 1557) | def _testNumpyInputFn(self, optimizer):
    method testNumpyInputFnStringOptimizer (line 1593) | def testNumpyInputFnStringOptimizer(self):
    method testNumpyInputFnOptimizerInstance (line 1596) | def testNumpyInputFnOptimizerInstance(self):
    method testParseExampleInputFn (line 1599) | def testParseExampleInputFn(self):
  function _rnn_classifier_fn (line 1655) | def _rnn_classifier_fn(feature_columns, n_classes, cell_units, model_dir,
  class RNNClassifierIntegrationTest (line 1666) | class RNNClassifierIntegrationTest(BaseRNNClassificationIntegrationTest,
    method __init__ (line 1669) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  function _rnn_classifier_dropout_fn (line 1674) | def _rnn_classifier_dropout_fn(feature_columns, n_classes, cell_units,
  class RNNClassifierDropoutIntegrationTest (line 1692) | class RNNClassifierDropoutIntegrationTest(BaseRNNClassificationIntegrati...
    method __init__ (line 1695) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  function _rnn_estimator_fn (line 1701) | def _rnn_estimator_fn(feature_columns, n_classes, cell_units, model_dir,
  class RNNEstimatorIntegrationTest (line 1712) | class RNNEstimatorIntegrationTest(BaseRNNClassificationIntegrationTest,
    method __init__ (line 1715) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class ModelFnTest (line 1721) | class ModelFnTest(tf.test.TestCase):
    method _test_sequential_mask_in_head (line 1724) | def _test_sequential_mask_in_head(self, mask=None):
    method testSequentialMaskInHead (line 1759) | def testSequentialMaskInHead(self):
    method testSequentialMaskInHeadWithMasks (line 1762) | def testSequentialMaskInHeadWithMasks(self):

FILE: tensorflow_estimator/python/estimator/canned/saved_model_estimator.py
  class SavedModelEstimator (line 35) | class SavedModelEstimator(estimator_lib.EstimatorV2):
    method __init__ (line 121) | def __init__(self, saved_model_dir, model_dir=None):
    method _extract_available_modes (line 148) | def _extract_available_modes(self):
    method _validate_mode (line 167) | def _validate_mode(self, mode):
    method _get_meta_graph_def_for_mode (line 174) | def _get_meta_graph_def_for_mode(self, mode):
    method _get_signature_def_for_mode (line 178) | def _get_signature_def_for_mode(self, mode):
    method _get_saver_def_from_mode (line 191) | def _get_saver_def_from_mode(self, mode):
    method _create_and_assert_global_step (line 195) | def _create_and_assert_global_step(self, graph):
    method _model_fn_from_saved_model (line 205) | def _model_fn_from_saved_model(self, features, labels, mode):
    method _restore_from_saver (line 290) | def _restore_from_saver(self, scaffold, session):
    method latest_checkpoint (line 294) | def latest_checkpoint(self):
  function _get_saved_model_ckpt (line 305) | def _get_saved_model_ckpt(saved_model_dir):
  function _clear_saved_model_collections (line 316) | def _clear_saved_model_collections():
  function _generate_input_map (line 330) | def _generate_input_map(signature_def, features, labels):
  function _check_same_dtype_and_shape (line 382) | def _check_same_dtype_and_shape(tensor, tensor_info, name):
  function _extract_eval_metrics (line 408) | def _extract_eval_metrics(output_dict):
  function _validate_and_extract_outputs (line 447) | def _validate_and_extract_outputs(mode, output_dict, method_name):

FILE: tensorflow_estimator/python/estimator/canned/saved_model_estimator_test.py
  function dummy_input_fn (line 40) | def dummy_input_fn():
  function _serving_feature_dict (line 46) | def _serving_feature_dict():
  function dummy_input_fn_features_only (line 50) | def dummy_input_fn_features_only():
  function dummy_supervised_receiver_fn (line 55) | def dummy_supervised_receiver_fn():
  function dummy_serving_receiver_fn (line 60) | def dummy_serving_receiver_fn():
  function model_fn_diff_modes (line 64) | def model_fn_diff_modes(features, labels, mode):
  function model_fn_with_trackable (line 93) | def model_fn_with_trackable(features, labels, mode):
  class SavedModelEstimatorTest (line 115) | class SavedModelEstimatorTest(tf.test.TestCase):
    method setUp (line 117) | def setUp(self):
    method tearDown (line 121) | def tearDown(self):
    method _get_tmp_dir (line 128) | def _get_tmp_dir(self):
    method _export_estimator (line 133) | def _export_estimator(self,
    method test_load_all_modes (line 154) | def test_load_all_modes(self):
    method test_load_all_modes_no_train (line 171) | def test_load_all_modes_no_train(self):
    method test_partial_exported_estimator (line 183) | def test_partial_exported_estimator(self):
    method test_with_incorrect_input (line 199) | def test_with_incorrect_input(self):
    method test_input_fn_with_global_step (line 219) | def test_input_fn_with_global_step(self):
    method test_re_export_saved_model_serving_only (line 233) | def test_re_export_saved_model_serving_only(self):
    method test_re_export_saved_model (line 256) | def test_re_export_saved_model(self):
    method test_re_export_saved_model_with_trackable (line 303) | def test_re_export_saved_model_with_trackable(self):
    method test_load_saved_model_from_serving_only (line 350) | def test_load_saved_model_from_serving_only(self):
    method test_with_local_init_op (line 384) | def test_with_local_init_op(self):
    method test_with_assets (line 411) | def test_with_assets(self):
    method test_with_working_input_fn (line 454) | def test_with_working_input_fn(self):
    method test_control_dependency (line 478) | def test_control_dependency(self):
    method test_saveable_resources (line 496) | def test_saveable_resources(self):

FILE: tensorflow_estimator/python/estimator/canned/timeseries/ar_model.py
  class LSTMPredictionModel (line 33) | class LSTMPredictionModel(tf_keras.models.Model):
    method __init__ (line 41) | def __init__(self,
    method call (line 65) | def call(self, input_window_features, output_window_features):
  class ARModel (line 77) | class ARModel(model.TimeSeriesModel):
    method __init__ (line 128) | def __init__(self,
    method initialize_graph (line 202) | def initialize_graph(self, input_statistics=None):
    method get_start_state (line 216) | def get_start_state(self):
    method random_model_parameters (line 227) | def random_model_parameters(self, seed=None):
    method generate (line 230) | def generate(self,
    method _predicted_covariance_op (line 237) | def _predicted_covariance_op(self, activations, num_values):
    method _predicted_mean_op (line 258) | def _predicted_mean_op(self, activations):
    method prediction_ops (line 269) | def prediction_ops(self, times, values, exogenous_regressors):
    method _output_window_predictions (line 342) | def _output_window_predictions(self, input_window_features,
    method loss_op (line 352) | def loss_op(self, targets, prediction_ops):
    method _process_exogenous_features (line 367) | def _process_exogenous_features(self, times, features):
    method predict (line 381) | def predict(self, features):
    method _process_window (line 555) | def _process_window(self, features, mode, exogenous_regressors):
    method get_batch_loss (line 617) | def get_batch_loss(self, features, mode, state):
    method _compute_time_features (line 788) | def _compute_time_features(self, time):

FILE: tensorflow_estimator/python/estimator/canned/timeseries/ar_model_test.py
  class ARModelTest (line 32) | class ARModelTest(tf.test.TestCase):
    method test_wrong_window_size (line 34) | def test_wrong_window_size(self):
    method test_predictions_direct_lstm (line 60) | def test_predictions_direct_lstm(self):
    method test_long_eval (line 77) | def test_long_eval(self):
    method test_long_eval_discard_indivisible (line 102) | def test_long_eval_discard_indivisible(self):

FILE: tensorflow_estimator/python/estimator/canned/timeseries/ar_model_training_test.py
  class InputFnBuilder (line 31) | class InputFnBuilder(object):
    method __init__ (line 33) | def __init__(self,
    method initialize_data_with_properties (line 46) | def initialize_data_with_properties(self, noise_stddev, periods, num_s...
    method train_or_test_input_fn (line 70) | def train_or_test_input_fn(self, time, data):
    method train_input_fn (line 86) | def train_input_fn(self):
    method test_input_fn (line 90) | def test_input_fn(self):
    method prediction_input_fn (line 94) | def prediction_input_fn(self):
    method true_values (line 125) | def true_values(self):
  class ARModelTrainingTest (line 134) | class ARModelTrainingTest(tf.test.TestCase):
    method train_helper (line 136) | def train_helper(self, input_window_size, loss, max_loss=None, periods...
    method test_autoregression_squared (line 199) | def test_autoregression_squared(self):
    method test_autoregression_short_input_window (line 202) | def test_autoregression_short_input_window(self):
    method test_autoregression_normal (line 205) | def test_autoregression_normal(self):
    method test_autoregression_normal_multiple_periods (line 211) | def test_autoregression_normal_multiple_periods(self):

FILE: tensorflow_estimator/python/estimator/canned/timeseries/estimators.py
  class TimeSeriesRegressor (line 33) | class TimeSeriesRegressor(estimator_lib.Estimator):
    method __init__ (line 36) | def __init__(self,
    method _model_start_state_placeholders (line 75) | def _model_start_state_placeholders(self,
    method build_one_shot_parsing_serving_input_receiver_fn (line 112) | def build_one_shot_parsing_serving_input_receiver_fn(self,
    method build_raw_serving_input_receiver_fn (line 210) | def build_raw_serving_input_receiver_fn(self,
  class LSTMAutoRegressor (line 293) | class LSTMAutoRegressor(TimeSeriesRegressor):
    method __init__ (line 373) | def __init__(self,

FILE: tensorflow_estimator/python/estimator/canned/timeseries/estimators_test.py
  class _SeedRunConfig (line 33) | class _SeedRunConfig(estimator_lib.RunConfig):
    method tf_random_seed (line 36) | def tf_random_seed(self):
  function _generate_data (line 40) | def _generate_data():
  function _build_input_fn_with_seed (line 47) | def _build_input_fn_with_seed(seed):
  class TimeSeriesRegressorTest (line 71) | class TimeSeriesRegressorTest(tf.test.TestCase):
    method _fit_restore_fit_test_template (line 73) | def _fit_restore_fit_test_template(self, estimator_fn, test_saved_model):
    method disabled_test_time_series_regressor (line 179) | def disabled_test_time_series_regressor(self):
    method test_ar_lstm_regressor (line 196) | def test_ar_lstm_regressor(self):

FILE: tensorflow_estimator/python/estimator/canned/timeseries/feature_keys.py
  class State (line 24) | class State(object):
  class Times (line 34) | class Times(object):
  class Values (line 40) | class Values(object):
  class TrainEvalFeatures (line 46) | class TrainEvalFeatures(Times, Values):
  class PredictionFeatures (line 51) | class PredictionFeatures(Times, State):
  class FilteringFeatures (line 56) | class FilteringFeatures(Times, Values, State):
  class PredictionResults (line 61) | class PredictionResults(Times):
  class FilteringResults (line 66) | class FilteringResults(Times, State):
  class SavedModelLabels (line 71) | class SavedModelLabels(object):

FILE: tensorflow_estimator/python/estimator/canned/timeseries/head.py
  class _NoStatePredictOutput (line 30) | class _NoStatePredictOutput(export_lib.PredictOutput):
    method as_signature_def (line 32) | def as_signature_def(self, receiver_tensors):
  class TimeSeriesRegressionHead (line 43) | class TimeSeriesRegressionHead(head_lib._Head):  # pylint:disable=protec...
    method __init__ (line 46) | def __init__(self,
    method name (line 68) | def name(self):
    method create_loss (line 73) | def create_loss(self, features, mode, logits=None, labels=None):
    method logits_dimension (line 82) | def logits_dimension(self):
    method _train_ops (line 86) | def _train_ops(self, features):
    method _evaluate_ops (line 100) | def _evaluate_ops(self, features):
    method _predict_ops (line 124) | def _predict_ops(self, features):
    method _serving_ops (line 133) | def _serving_ops(self, features):
    method _convert_feature_to_tensor (line 166) | def _convert_feature_to_tensor(self, name, value):
    method _gather_state (line 179) | def _gather_state(self, features):
    method _check_predict_features (line 199) | def _check_predict_features(self, features):
    method create_estimator_spec (line 222) | def create_estimator_spec(self, features, mode, labels=None):
  class OneShotPredictionHead (line 271) | class OneShotPredictionHead(TimeSeriesRegressionHead):
    method _check_predict_features (line 288) | def _check_predict_features(self, features):
    method _evaluate_ops (line 318) | def _evaluate_ops(self, features):
    method _serving_ops (line 326) | def _serving_ops(self, features):
  function _check_feature_shapes_compatible_with (line 360) | def _check_feature_shapes_compatible_with(features,
  function _check_train_eval_features (line 391) | def _check_train_eval_features(features, model):
  function _identity_metric_single (line 424) | def _identity_metric_single(name, input_tensor):
  function _identity_metric_nested (line 453) | def _identity_metric_nested(name, input_tensors):
  function state_to_dictionary (line 466) | def state_to_dictionary(state_tuple):

FILE: tensorflow_estimator/python/estimator/canned/timeseries/head_test.py
  class HeadTest (line 41) | class HeadTest(tf.test.TestCase):
    method test_labels_provided_error (line 43) | def test_labels_provided_error(self):
    method test_unknown_mode (line 55) | def test_unknown_mode(self):
  class _TickerModel (line 61) | class _TickerModel(object):
    method initialize_graph (line 65) | def initialize_graph(self, input_statistics):
    method define_loss (line 68) | def define_loss(self, features, mode):
  class EvaluationMetricsTests (line 78) | class EvaluationMetricsTests(tf.test.TestCase):
    method test_metrics_consistent (line 80) | def test_metrics_consistent(self):
    method test_custom_metrics (line 131) | def test_custom_metrics(self):
  class _StubModel (line 169) | class _StubModel(object):
    method initialize_graph (line 173) | def initialize_graph(self, input_statistics):
  function _stub_model_fn (line 177) | def _stub_model_fn():
  class TrainEvalFeatureCheckingTests (line 184) | class TrainEvalFeatureCheckingTests(tf.test.TestCase):
    method test_no_time_feature (line 186) | def test_no_time_feature(self):
    method test_no_value_feature (line 197) | def test_no_value_feature(self):
    method test_bad_time_rank (line 208) | def test_bad_time_rank(self):
    method test_bad_value_rank (line 222) | def test_bad_value_rank(self):
    method test_bad_value_num_features (line 236) | def test_bad_value_num_features(self):
    method test_bad_exogenous_shape (line 250) | def test_bad_exogenous_shape(self):
  class PredictFeatureCheckingTests (line 265) | class PredictFeatureCheckingTests(tf.test.TestCase):
    method test_no_time_feature (line 267) | def test_no_time_feature(self):
    method test_no_start_state_feature (line 279) | def test_no_start_state_feature(self):
    method test_bad_time_rank (line 289) | def test_bad_time_rank(self):
    method test_bad_exogenous_shape (line 302) | def test_bad_exogenous_shape(self):
  class OneShotTests (line 317) | class OneShotTests(parameterized.TestCase):
    method test_one_shot_prediction_head_export (line 319) | def test_one_shot_prediction_head_export(self):

FILE: tensorflow_estimator/python/estimator/canned/timeseries/math_utils.py
  function replicate_state (line 27) | def replicate_state(start_state, batch_size):
  class InputStatisticsFromMiniBatch (line 92) | class InputStatisticsFromMiniBatch(object):
    method __init__ (line 95) | def __init__(self, num_features, dtype, starting_variance_window_size=...
    method initialize_graph (line 108) | def initialize_graph(self, features, update_statistics=True):
    class _AdaptiveInputAuxiliaryStatistics (line 166) | class _AdaptiveInputAuxiliaryStatistics(
      method __new__ (line 189) | def __new__(cls, num_features, dtype):
    method _update_statistics_from_mini_batch (line 230) | def _update_statistics_from_mini_batch(self, statistics, auxiliary_var...
    method _create_variable_statistics_object (line 357) | def _create_variable_statistics_object(self):

FILE: tensorflow_estimator/python/estimator/canned/timeseries/math_utils_test.py
  class InputStatisticsTests (line 26) | class InputStatisticsTests(tf.test.TestCase):
    method _input_statistics_test_template (line 28) | def _input_statistics_test_template(self,
    method test_queue (line 84) | def test_queue(self):

FILE: tensorflow_estimator/python/estimator/canned/timeseries/model.py
  class TimeSeriesModel (line 44) | class TimeSeriesModel(object):
    method __init__ (line 47) | def __init__(self,
    method exogenous_feature_columns (line 74) | def exogenous_feature_columns(self):
    method generate (line 81) | def generate(self,
    method initialize_graph (line 111) | def initialize_graph(self, input_statistics=None):
    method _scale_data (line 126) | def _scale_data(self, data):
    method _scale_variance (line 133) | def _scale_variance(self, variance):
    method _scale_back_data (line 140) | def _scale_back_data(self, data):
    method _scale_back_variance (line 147) | def _scale_back_variance(self, variance):
    method _check_graph_initialized (line 154) | def _check_graph_initialized(self):
    method define_loss (line 161) | def define_loss(self, features, mode):
    method get_start_state (line 198) | def get_start_state(self):
    method get_batch_loss (line 208) | def get_batch_loss(self, features, mode, state):
    method predict (line 225) | def predict(self, features):
    method _get_exogenous_embedding_shape (line 251) | def _get_exogenous_embedding_shape(self):
    method _process_exogenous_features (line 272) | def _process_exogenous_features(self, times, features):

FILE: tensorflow_estimator/python/estimator/canned/timeseries/model_utils.py
  function fully_connected (line 27) | def fully_connected(inp,
  function canonicalize_times_or_steps_from_output (line 47) | def canonicalize_times_or_steps_from_output(times, steps,

FILE: tensorflow_estimator/python/estimator/canned/timeseries/saved_model_utils.py
  function _canonicalize_numpy_data (line 34) | def _canonicalize_numpy_data(data, require_single_batch):
  function _colate_features_to_feeds_and_fetches (line 120) | def _colate_features_to_feeds_and_fetches(signature,
  function predict_continuation (line 149) | def predict_continuation(continue_from,
  function cold_start_filter (line 207) | def cold_start_filter(signatures, session, features):
  function filter_continuation (line 251) | def filter_continuation(continue_from, signatures, session, features):

FILE: tensorflow_estimator/python/estimator/canned/timeseries/state_management.py
  class PassthroughStateManager (line 26) | class PassthroughStateManager(object):
    method __init__ (line 29) | def __init__(self):
    method initialize_graph (line 33) | def initialize_graph(self, model, input_statistics=None):
    method define_loss (line 39) | def define_loss(self, model, features, mode):
  class _OverridableStateManager (line 62) | class _OverridableStateManager(PassthroughStateManager):
    method _define_loss_with_saved_state (line 66) | def _define_loss_with_saved_state(self, model, features, mode):
    method define_loss (line 69) | def define_loss(self, model, features, mode):
  class FilteringOnlyStateManager (line 88) | class FilteringOnlyStateManager(_OverridableStateManager):
    method _define_loss_with_saved_state (line 97) | def _define_loss_with_saved_state(self, model, features, mode):

FILE: tensorflow_estimator/python/estimator/canned/v1/baseline_estimator_test_v1.py
  function assert_close (line 40) | def assert_close(expected, actual, rtol=1e-04, name='assert_close'):
  function save_variables_to_ckpt (line 55) | def save_variables_to_ckpt(model_dir):
  function _baseline_estimator_fn (line 62) | def _baseline_estimator_fn(weight_column=None, label_dimension=1, **kwar...
  class BaselineEstimatorEvaluationTest (line 72) | class BaselineEstimatorEvaluationTest(tf.test.TestCase):
    method setUp (line 74) | def setUp(self):
    method tearDown (line 77) | def tearDown(self):
    method test_evaluation_batch (line 82) | def test_evaluation_batch(self):
    method test_evaluation_weights (line 109) | def test_evaluation_weights(self):
    method test_evaluation_for_multi_dimensions (line 139) | def test_evaluation_for_multi_dimensions(self):
  class BaselineEstimatorPredictTest (line 169) | class BaselineEstimatorPredictTest(tf.test.TestCase):
    method setUp (line 171) | def setUp(self):
    method tearDown (line 174) | def tearDown(self):
    method test_1d (line 179) | def test_1d(self):
    method testMultiDim (line 199) | def testMultiDim(self):
  class BaselineEstimatorIntegrationTest (line 226) | class BaselineEstimatorIntegrationTest(tf.test.TestCase):
    method setUp (line 228) | def setUp(self):
    method tearDown (line 231) | def tearDown(self):
    method _test_complete_flow (line 236) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method test_numpy_input_fn (line 267) | def test_numpy_input_fn(self):
  class BaselineEstimatorTrainingTest (line 305) | class BaselineEstimatorTrainingTest(tf.test.TestCase):
    method setUp (line 307) | def setUp(self):
    method tearDown (line 310) | def tearDown(self):
    method _mock_optimizer (line 315) | def _mock_optimizer(self, expected_loss=None):
    method _assert_checkpoint (line 350) | def _assert_checkpoint(self,
    method testFromScratch (line 370) | def testFromScratch(self):
    method testFromCheckpoint (line 388) | def testFromCheckpoint(self):

FILE: tensorflow_estimator/python/estimator/canned/v1/baseline_test_v1.py
  function assert_close (line 56) | def assert_close(expected, actual, rtol=1e-04, name='assert_close'):
  function save_variables_to_ckpt (line 71) | def save_variables_to_ckpt(model_dir):
  function queue_parsed_features (line 78) | def queue_parsed_features(feature_map):
  function sorted_key_dict (line 93) | def sorted_key_dict(unsorted_dict):
  function sigmoid (line 97) | def sigmoid(x):
  function _baseline_regressor_fn (line 101) | def _baseline_regressor_fn(*args, **kwargs):
  function _baseline_classifier_fn (line 105) | def _baseline_classifier_fn(*args, **kwargs):
  class BaselineRegressorEvaluationTest (line 114) | class BaselineRegressorEvaluationTest(tf.test.TestCase):
    method setUp (line 116) | def setUp(self):
    method tearDown (line 119) | def tearDown(self):
    method test_evaluation_for_simple_data (line 124) | def test_evaluation_for_simple_data(self):
    method test_evaluation_batch (line 147) | def test_evaluation_batch(self):
    method test_evaluation_weights (line 174) | def test_evaluation_weights(self):
    method test_evaluation_for_multi_dimensions (line 204) | def test_evaluation_for_multi_dimensions(self):
  class BaselineRegressorPredictTest (line 234) | class BaselineRegressorPredictTest(tf.test.TestCase):
    method setUp (line 236) | def setUp(self):
    method tearDown (line 239) | def tearDown(self):
    method test_1d (line 244) | def test_1d(self):
    method testMultiDim (line 264) | def testMultiDim(self):
  class BaselineRegressorIntegrationTest (line 291) | class BaselineRegressorIntegrationTest(tf.test.TestCase):
    method setUp (line 293) | def setUp(self):
    method tearDown (line 296) | def tearDown(self):
    method _test_complete_flow (line 301) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method test_numpy_input_fn (line 332) | def test_numpy_input_fn(self):
    method test_pandas_input_fn (line 368) | def test_pandas_input_fn(self):
    method test_input_fn_from_parse_example (line 397) | def test_input_fn_from_parse_example(self):
  class BaselineRegressorTrainingTest (line 459) | class BaselineRegressorTrainingTest(tf.test.TestCase):
    method setUp (line 461) | def setUp(self):
    method tearDown (line 464) | def tearDown(self):
    method _mock_optimizer (line 469) | def _mock_optimizer(self, expected_loss=None):
    method _assert_checkpoint (line 504) | def _assert_checkpoint(self,
    method testFromScratchWithDefaultOptimizer (line 524) | def testFromScratchWithDefaultOptimizer(self):
    method testTrainWithOneDimLabel (line 536) | def testTrainWithOneDimLabel(self):
    method testTrainWithOneDimWeight (line 553) | def testTrainWithOneDimWeight(self):
    method testFromScratch (line 576) | def testFromScratch(self):
    method testFromCheckpoint (line 594) | def testFromCheckpoint(self):
    method testFromCheckpointMultiBatch (line 623) | def testFromCheckpointMultiBatch(self):
  class BaselineClassifierTrainingTest (line 660) | class BaselineClassifierTrainingTest(tf.test.TestCase):
    method setUp (line 662) | def setUp(self):
    method tearDown (line 665) | def tearDown(self):
    method _mock_optimizer (line 669) | def _mock_optimizer(self, expected_loss=None):
    method _assert_checkpoint (line 700) | def _assert_checkpoint(self,
    method _testFromScratchWithDefaultOptimizer (line 722) | def _testFromScratchWithDefaultOptimizer(self, n_classes):
    method testBinaryClassesFromScratchWithDefaultOptimizer (line 734) | def testBinaryClassesFromScratchWithDefaultOptimizer(self):
    method testMultiClassesFromScratchWithDefaultOptimizer (line 737) | def testMultiClassesFromScratchWithDefaultOptimizer(self):
    method _testTrainWithTwoDimsLabel (line 740) | def _testTrainWithTwoDimsLabel(self, n_classes):
    method testBinaryClassesTrainWithTwoDimsLabel (line 759) | def testBinaryClassesTrainWithTwoDimsLabel(self):
    method testMultiClassesTrainWithTwoDimsLabel (line 762) | def testMultiClassesTrainWithTwoDimsLabel(self):
    method _testTrainWithOneDimLabel (line 765) | def _testTrainWithOneDimLabel(self, n_classes):
    method testBinaryClassesTrainWithOneDimLabel (line 782) | def testBinaryClassesTrainWithOneDimLabel(self):
    method testMultiClassesTrainWithOneDimLabel (line 785) | def testMultiClassesTrainWithOneDimLabel(self):
    method _testTrainWithTwoDimsWeight (line 788) | def _testTrainWithTwoDimsWeight(self, n_classes):
    method testBinaryClassesTrainWithTwoDimsWeight (line 810) | def testBinaryClassesTrainWithTwoDimsWeight(self):
    method testMultiClassesTrainWithTwoDimsWeight (line 813) | def testMultiClassesTrainWithTwoDimsWeight(self):
    method _testTrainWithOneDimWeight (line 816) | def _testTrainWithOneDimWeight(self, n_classes):
    method testBinaryClassesTrainWithOneDimWeight (line 836) | def testBinaryClassesTrainWithOneDimWeight(self):
    method testMultiClassesTrainWithOneDimWeight (line 839) | def testMultiClassesTrainWithOneDimWeight(self):
    method _testFromScratch (line 842) | def _testFromScratch(self, n_classes):
    method testBinaryClassesFromScratch (line 874) | def testBinaryClassesFromScratch(self):
    method testMultiClassesFromScratch (line 877) | def testMultiClassesFromScratch(self):
    method _testFromCheckpoint (line 880) | def _testFromCheckpoint(self, n_classes):
    method testBinaryClassesFromCheckpoint (line 928) | def testBinaryClassesFromCheckpoint(self):
    method testMultiClassesFromCheckpoint (line 931) | def testMultiClassesFromCheckpoint(self):
    method _testFromCheckpointFloatLabels (line 934) | def _testFromCheckpointFloatLabels(self, n_classes):
    method testBinaryClassesFromCheckpointFloatLabels (line 968) | def testBinaryClassesFromCheckpointFloatLabels(self):
    method testMultiClassesFromCheckpointFloatLabels (line 971) | def testMultiClassesFromCheckpointFloatLabels(self):
    method _testFromCheckpointMultiBatch (line 974) | def _testFromCheckpointMultiBatch(self, n_classes):
    method testBinaryClassesFromCheckpointMultiBatch (line 1031) | def testBinaryClassesFromCheckpointMultiBatch(self):
    method testMultiClassesFromCheckpointMultiBatch (line 1034) | def testMultiClassesFromCheckpointMultiBatch(self):
  class BaselineClassifierEvaluationTest (line 1039) | class BaselineClassifierEvaluationTest(tf.test.TestCase):
    method setUp (line 1041) | def setUp(self):
    method tearDown (line 1044) | def tearDown(self):
    method _test_evaluation_for_simple_data (line 1048) | def _test_evaluation_for_simple_data(self, n_classes):
    method test_binary_classes_evaluation_for_simple_data (line 1100) | def test_binary_classes_evaluation_for_simple_data(self):
    method test_multi_classes_evaluation_for_simple_data (line 1103) | def test_multi_classes_evaluation_for_simple_data(self):
    method _test_evaluation_batch (line 1106) | def _test_evaluation_batch(self, n_classes):
    method test_binary_classes_evaluation_batch (line 1168) | def test_binary_classes_evaluation_batch(self):
    method test_multi_classes_evaluation_batch (line 1171) | def test_multi_classes_evaluation_batch(self):
    method _test_evaluation_weights (line 1174) | def _test_evaluation_weights(self, n_classes):
    method test_binary_classes_evaluation_weights (line 1250) | def test_binary_classes_evaluation_weights(self):
    method test_multi_classes_evaluation_weights (line 1253) | def test_multi_classes_evaluation_weights(self):
  class BaselineClassifierPredictTest (line 1258) | class BaselineClassifierPredictTest(tf.test.TestCase):
    method setUp (line 1260) | def setUp(self):
    method tearDown (line 1263) | def tearDown(self):
    method _testPredictions (line 1267) | def _testPredictions(self, n_classes, label_vocabulary, label_output_fn):
    method testBinaryClassesWithoutLabelVocabulary (line 1334) | def testBinaryClassesWithoutLabelVocabulary(self):
    method testBinaryClassesWithLabelVocabulary (line 1341) | def testBinaryClassesWithLabelVocabulary(self):
    method testMultiClassesWithoutLabelVocabulary (line 1348) | def testMultiClassesWithoutLabelVocabulary(self):
    method testMultiClassesWithLabelVocabulary (line 1355) | def testMultiClassesWithLabelVocabulary(self):
  class BaselineClassifierIntegrationTest (line 1364) | class BaselineClassifierIntegrationTest(tf.test.TestCase):
    method setUp (line 1366) | def setUp(self):
    method tearDown (line 1369) | def tearDown(self):
    method _test_complete_flow (line 1373) | def _test_complete_flow(self, n_classes, train_input_fn, eval_input_fn,
    method _test_numpy_input_fn (line 1404) | def _test_numpy_input_fn(self, n_classes):
    method test_binary_classes_numpy_input_fn (line 1440) | def test_binary_classes_numpy_input_fn(self):
    method test_multi_classes_numpy_input_fn (line 1443) | def test_multi_classes_numpy_input_fn(self):
    method _test_pandas_input_fn (line 1446) | def _test_pandas_input_fn(self, n_classes):
    method test_binary_classes_pandas_input_fn (line 1475) | def test_binary_classes_pandas_input_fn(self):
    method test_multi_classes_pandas_input_fn (line 1478) | def test_multi_classes_pandas_input_fn(self):
    method _test_input_fn_from_parse_example (line 1481) | def _test_input_fn_from_parse_example(self, n_classes):
    method test_binary_classes_input_fn_from_parse_example (line 1540) | def test_binary_classes_input_fn_from_parse_example(self):
    method test_multi_classes_input_fn_from_parse_example (line 1543) | def test_multi_classes_input_fn_from_parse_example(self):
  class BaselineLogitFnTest (line 1551) | class BaselineLogitFnTest(tf.test.TestCase):
    method test_basic_logit_correctness (line 1553) | def test_basic_logit_correctness(self):

FILE: tensorflow_estimator/python/estimator/canned/v1/dnn_estimator_test_v1.py
  function _dnn_estimator_fn (line 36) | def _dnn_estimator_fn(weight_column=None, label_dimension=1, **kwargs):
  function _dnn_estimator_classifier_fn (line 47) | def _dnn_estimator_classifier_fn(n_classes=3, **kwargs):
  class DNNEstimatorEvaluateTest (line 55) | class DNNEstimatorEvaluateTest(
    method __init__ (line 58) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNEstimatorPredictTest (line 65) | class DNNEstimatorPredictTest(dnn_testing_utils_v1.BaseDNNRegressorPredi...
    method __init__ (line 68) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNEstimatorTrainTest (line 75) | class DNNEstimatorTrainTest(dnn_testing_utils_v1.BaseDNNRegressorTrainTest,
    method __init__ (line 78) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNEstimatorWarmStartingTest (line 85) | class DNNEstimatorWarmStartingTest(dnn_testing_utils_v1.BaseDNNWarmStart...
    method __init__ (line 88) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNEstimatorIntegrationTest (line 95) | class DNNEstimatorIntegrationTest(tf.test.TestCase):
    method setUp (line 97) | def setUp(self):
    method tearDown (line 100) | def tearDown(self):
    method _test_complete_flow (line 105) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method test_numpy_input_fn (line 141) | def test_numpy_input_fn(self):

FILE: tensorflow_estimator/python/estimator/canned/v1/dnn_linear_combined_estimator_test_v1.py
  function _dnn_only_estimator_fn (line 38) | def _dnn_only_estimator_fn(hidden_units,
  class DNNOnlyEstimatorEvaluateTest (line 65) | class DNNOnlyEstimatorEvaluateTest(
    method __init__ (line 68) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyEstimatorPredictTest (line 75) | class DNNOnlyEstimatorPredictTest(
    method __init__ (line 78) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyEstimatorTrainTest (line 85) | class DNNOnlyEstimatorTrainTest(dnn_testing_utils_v1.BaseDNNRegressorTra...
    method __init__ (line 88) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  function _linear_only_estimator_fn (line 94) | def _linear_only_estimator_fn(feature_columns,
  class LinearOnlyEstimatorEvaluateTest (line 117) | class LinearOnlyEstimatorEvaluateTest(
    method __init__ (line 121) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyEstimatorPredictTest (line 128) | class LinearOnlyEstimatorPredictTest(
    method __init__ (line 131) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyEstimatorTrainTest (line 138) | class LinearOnlyEstimatorTrainTest(
    method __init__ (line 141) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNLinearCombinedEstimatorIntegrationTest (line 148) | class DNNLinearCombinedEstimatorIntegrationTest(tf.test.TestCase):
    method setUp (line 150) | def setUp(self):
    method tearDown (line 153) | def tearDown(self):
    method _test_complete_flow (line 158) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method test_numpy_input_fn (line 199) | def test_numpy_input_fn(self):

FILE: tensorflow_estimator/python/estimator/canned/v1/dnn_linear_combined_test_v1.py
  class DNNOnlyModelFnTest (line 65) | class DNNOnlyModelFnTest(dnn_testing_utils_v1.BaseDNNModelFnTest,
    method __init__ (line 68) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
    method _dnn_only_model_fn (line 73) | def _dnn_only_model_fn(self,
  function _linear_regressor_fn (line 101) | def _linear_regressor_fn(feature_columns,
  class LinearOnlyRegressorPartitionerTest (line 121) | class LinearOnlyRegressorPartitionerTest(
    method __init__ (line 125) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyRegressorPartitionerV2Test (line 132) | class LinearOnlyRegressorPartitionerV2Test(
    method __init__ (line 136) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyRegressorEvaluationTest (line 143) | class LinearOnlyRegressorEvaluationTest(
    method __init__ (line 147) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyRegressorEvaluationV2Test (line 154) | class LinearOnlyRegressorEvaluationV2Test(
    method __init__ (line 158) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyRegressorPredictTest (line 165) | class LinearOnlyRegressorPredictTest(
    method __init__ (line 168) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyRegressorPredictV2Test (line 175) | class LinearOnlyRegressorPredictV2Test(
    method __init__ (line 178) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyRegressorIntegrationTest (line 185) | class LinearOnlyRegressorIntegrationTest(
    method __init__ (line 189) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyRegressorIntegrationV2Test (line 196) | class LinearOnlyRegressorIntegrationV2Test(
    method __init__ (line 200) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyRegressorTrainingTest (line 207) | class LinearOnlyRegressorTrainingTest(
    method __init__ (line 210) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyRegressorTrainingV2Test (line 217) | class LinearOnlyRegressorTrainingV2Test(
    method __init__ (line 220) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  function _linear_classifier_fn (line 226) | def _linear_classifier_fn(feature_columns,
  class LinearOnlyClassifierTrainingTest (line 248) | class LinearOnlyClassifierTrainingTest(
    method __init__ (line 251) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyClassifierTrainingV2Test (line 258) | class LinearOnlyClassifierTrainingV2Test(
    method __init__ (line 261) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyClassifierClassesEvaluationTest (line 270) | class LinearOnlyClassifierClassesEvaluationTest(
    method __init__ (line 274) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyClassifierClassesEvaluationV2Test (line 281) | class LinearOnlyClassifierClassesEvaluationV2Test(
    method __init__ (line 285) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyClassifierPredictTest (line 294) | class LinearOnlyClassifierPredictTest(
    method __init__ (line 297) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyClassifierPredictV2Test (line 304) | class LinearOnlyClassifierPredictV2Test(
    method __init__ (line 307) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyClassifierIntegrationTest (line 316) | class LinearOnlyClassifierIntegrationTest(
    method __init__ (line 320) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyClassifierIntegrationV2Test (line 327) | class LinearOnlyClassifierIntegrationV2Test(
    method __init__ (line 331) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNLinearCombinedRegressorIntegrationTest (line 341) | class DNNLinearCombinedRegressorIntegrationTest(tf.test.TestCase):
    method setUp (line 343) | def setUp(self):
    method tearDown (line 346) | def tearDown(self):
    method _test_complete_flow_helper (line 351) | def _test_complete_flow_helper(self, linear_feature_columns,
    method _test_complete_flow (line 386) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method _test_complete_flow_mix1 (line 403) | def _test_complete_flow_mix1(self, train_input_fn, eval_input_fn,
    method _test_complete_flow_mix2 (line 421) | def _test_complete_flow_mix2(self, train_input_fn, eval_input_fn,
    method _test_numpy_input_fn_helper (line 439) | def _test_numpy_input_fn_helper(self, fc_impl, fn_to_run):
    method test_numpy_input_fn_basic (line 466) | def test_numpy_input_fn_basic(self, fc_impl):
    method test_numpy_input_fn_mix1 (line 469) | def test_numpy_input_fn_mix1(self, fc_impl):
    method test_numpy_input_fn_mix2 (line 472) | def test_numpy_input_fn_mix2(self, fc_impl):
    method _test_pandas_input_fn_helper (line 475) | def _test_pandas_input_fn_helper(self, fc_impl, fn_to_run):
    method test_pandas_input_fn_basic (line 500) | def test_pandas_input_fn_basic(self, fc_impl):
    method test_pandas_input_fn_mix1 (line 503) | def test_pandas_input_fn_mix1(self, fc_impl):
    method test_pandas_input_fn_mix2 (line 506) | def test_pandas_input_fn_mix2(self, fc_impl):
    method _test_input_fn_from_parse_example_helper (line 509) | def _test_input_fn_from_parse_example_helper(self, fc_impl, fn_to_run):
    method test_input_fn_from_parse_example_basic (line 567) | def test_input_fn_from_parse_example_basic(self, fc_impl):
    method test_input_fn_from_parse_example_mix1 (line 571) | def test_input_fn_from_parse_example_mix1(self, fc_impl):
    method test_input_fn_from_parse_example_mix2 (line 575) | def test_input_fn_from_parse_example_mix2(self, fc_impl):
  function _dnn_classifier_fn (line 581) | def _dnn_classifier_fn(hidden_units,
  class DNNOnlyClassifierEvaluateTest (line 603) | class DNNOnlyClassifierEvaluateTest(
    method __init__ (line 606) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyClassifierEvaluateV2Test (line 613) | class DNNOnlyClassifierEvaluateV2Test(
    method __init__ (line 616) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyClassifierPredictTest (line 623) | class DNNOnlyClassifierPredictTest(
    method __init__ (line 626) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyClassifierPredictV2Test (line 633) | class DNNOnlyClassifierPredictV2Test(
    method __init__ (line 636) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyClassifierTrainTest (line 643) | class DNNOnlyClassifierTrainTest(
    method __init__ (line 646) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyClassifierTrainV2Test (line 653) | class DNNOnlyClassifierTrainV2Test(
    method __init__ (line 656) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  function _dnn_regressor_fn (line 663) | def _dnn_regressor_fn(hidden_units,
  class DNNOnlyRegressorEvaluateTest (line 683) | class DNNOnlyRegressorEvaluateTest(
    method __init__ (line 686) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyRegressorEvaluateV2Test (line 693) | class DNNOnlyRegressorEvaluateV2Test(
    method __init__ (line 696) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyRegressorPredictTest (line 703) | class DNNOnlyRegressorPredictTest(
    method __init__ (line 706) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyRegressorPredictV2Test (line 713) | class DNNOnlyRegressorPredictV2Test(
    method __init__ (line 716) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyRegressorTrainTest (line 723) | class DNNOnlyRegressorTrainTest(dnn_testing_utils_v1.BaseDNNRegressorTra...
    method __init__ (line 726) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyRegressorTrainV2Test (line 733) | class DNNOnlyRegressorTrainV2Test(
    method __init__ (line 736) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNLinearCombinedClassifierIntegrationTest (line 744) | class DNNLinearCombinedClassifierIntegrationTest(tf.test.TestCase):
    method setUp (line 746) | def setUp(self):
    method tearDown (line 749) | def tearDown(self):
    method _as_label (line 754) | def _as_label(self, data_in_float):
    method _test_complete_flow (line 757) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method test_numpy_input_fn (line 798) | def test_numpy_input_fn(self, fc_impl):
    method test_pandas_input_fn (line 828) | def test_pandas_input_fn(self, fc_impl):
    method test_input_fn_from_parse_example (line 854) | def test_input_fn_from_parse_example(self, fc_impl):
  class DNNLinearCombinedTests (line 918) | class DNNLinearCombinedTests(tf.test.TestCase):
    method setUp (line 920) | def setUp(self):
    method tearDown (line 923) | def tearDown(self):
    method _mock_optimizer (line 927) | def _mock_optimizer(self, real_optimizer, var_name_prefix):
    method test_train_op_calls_both_dnn_and_linear (line 949) | def test_train_op_calls_both_dnn_and_linear(self, fc_impl):
    method test_dnn_and_linear_logits_are_added (line 974) | def test_dnn_and_linear_logits_are_added(self, fc_impl):
  class DNNLinearCombinedWarmStartingTest (line 1003) | class DNNLinearCombinedWarmStartingTest(tf.test.TestCase):
    method setUp (line 1005) | def setUp(self):
    method tearDown (line 1019) | def tearDown(self):
    method test_classifier_basic_warm_starting (line 1024) | def test_classifier_basic_warm_starting(self, fc_impl):
    method test_regressor_basic_warm_starting (line 1064) | def test_regressor_basic_warm_starting(self, fc_impl):
    method test_warm_starting_selective_variables (line 1102) | def test_warm_starting_selective_variables(self, fc_impl):

FILE: tensorflow_estimator/python/estimator/canned/v1/dnn_test_fc_v1_v1.py
  function _dnn_classifier_fn (line 53) | def _dnn_classifier_fn(*args, **kwargs):
  class DNNModelFnTest (line 58) | class DNNModelFnTest(dnn_testing_utils_v1.BaseDNNModelFnTest, tf.test.Te...
    method __init__ (line 60) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNLogitFnTest (line 67) | class DNNLogitFnTest(dnn_testing_utils_v1.BaseDNNLogitFnTest, tf.test.Te...
    method __init__ (line 69) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNWarmStartingTest (line 76) | class DNNWarmStartingTest(dnn_testing_utils_v1.BaseDNNWarmStartingTest,
    method __init__ (line 79) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNClassifierEvaluateTest (line 86) | class DNNClassifierEvaluateTest(
    method __init__ (line 89) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNClassifierPredictTest (line 96) | class DNNClassifierPredictTest(
    method __init__ (line 99) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNClassifierTrainTest (line 106) | class DNNClassifierTrainTest(dnn_testing_utils_v1.BaseDNNClassifierTrain...
    method __init__ (line 109) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  function _dnn_regressor_fn (line 115) | def _dnn_regressor_fn(*args, **kwargs):
  class DNNRegressorEvaluateTest (line 120) | class DNNRegressorEvaluateTest(
    method __init__ (line 123) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNRegressorPredictTest (line 130) | class DNNRegressorPredictTest(dnn_testing_utils_v1.BaseDNNRegressorPredi...
    method __init__ (line 133) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNRegressorTrainTest (line 140) | class DNNRegressorTrainTest(dnn_testing_utils_v1.BaseDNNRegressorTrainTest,
    method __init__ (line 143) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  function _queue_parsed_features (line 149) | def _queue_parsed_features(feature_map):
  class DNNRegressorIntegrationTest (line 165) | class DNNRegressorIntegrationTest(tf.test.TestCase, parameterized.TestCa...
    method setUp (line 167) | def setUp(self):
    method tearDown (line 170) | def tearDown(self):
    method _test_complete_flow (line 175) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method test_numpy_input_fn (line 212) | def test_numpy_input_fn(self):
    method test_pandas_input_fn (line 238) | def test_pandas_input_fn(self):
    method test_input_fn_from_parse_example (line 262) | def test_input_fn_from_parse_example(self):
  class DNNClassifierIntegrationTest (line 321) | class DNNClassifierIntegrationTest(tf.test.TestCase):
    method setUp (line 323) | def setUp(self):
    method tearDown (line 326) | def tearDown(self):
    method _as_label (line 331) | def _as_label(self, data_in_float):
    method _test_complete_flow (line 334) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method test_numpy_input_fn (line 371) | def test_numpy_input_fn(self):
    method test_pandas_input_fn (line 400) | def test_pandas_input_fn(self):
    method test_input_fn_from_parse_example (line 425) | def test_input_fn_from_parse_example(self):

FILE: tensorflow_estimator/python/estimator/canned/v1/dnn_test_fc_v2_v1.py
  function _dnn_classifier_fn (line 50) | def _dnn_classifier_fn(*args, **kwargs):
  class DNNModelFnV2Test (line 55) | class DNNModelFnV2Test(dnn_testing_utils_v1.BaseDNNModelFnTest,
    method __init__ (line 58) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNLogitFnV2Test (line 65) | class DNNLogitFnV2Test(dnn_testing_utils_v1.BaseDNNLogitFnTest,
    method __init__ (line 68) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNWarmStartingV2Test (line 75) | class DNNWarmStartingV2Test(dnn_testing_utils_v1.BaseDNNWarmStartingTest,
    method __init__ (line 78) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNClassifierEvaluateV2Test (line 85) | class DNNClassifierEvaluateV2Test(
    method __init__ (line 88) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNClassifierPredictV2Test (line 95) | class DNNClassifierPredictV2Test(
    method __init__ (line 98) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNClassifierTrainV2Test (line 105) | class DNNClassifierTrainV2Test(dnn_testing_utils_v1.BaseDNNClassifierTra...
    method __init__ (line 108) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  function _dnn_regressor_fn (line 114) | def _dnn_regressor_fn(*args, **kwargs):
  class DNNRegressorEvaluateV2Test (line 119) | class DNNRegressorEvaluateV2Test(
    method __init__ (line 122) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNRegressorPredictV2Test (line 129) | class DNNRegressorPredictV2Test(
    method __init__ (line 132) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNRegressorTrainV2Test (line 139) | class DNNRegressorTrainV2Test(dnn_testing_utils_v1.BaseDNNRegressorTrain...
    method __init__ (line 142) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  function _queue_parsed_features (line 148) | def _queue_parsed_features(feature_map):
  class DNNRegressorIntegrationTest (line 164) | class DNNRegressorIntegrationTest(tf.test.TestCase, parameterized.TestCa...
    method setUp (line 166) | def setUp(self):
    method tearDown (line 169) | def tearDown(self):
    method _test_complete_flow (line 174) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method test_numpy_input_fn (line 210) | def test_numpy_input_fn(self):
    method test_pandas_input_fn (line 236) | def test_pandas_input_fn(self):
    method test_input_fn_from_parse_example (line 260) | def test_input_fn_from_parse_example(self):
  class DNNClassifierIntegrationTest (line 319) | class DNNClassifierIntegrationTest(tf.test.TestCase):
    method setUp (line 321) | def setUp(self):
    method tearDown (line 324) | def tearDown(self):
    method _as_label (line 329) | def _as_label(self, data_in_float):
    method _test_complete_flow (line 332) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method test_numpy_input_fn (line 368) | def test_numpy_input_fn(self):
    method test_pandas_input_fn (line 397) | def test_pandas_input_fn(self):
    method test_input_fn_from_parse_example (line 422) | def test_input_fn_from_parse_example(self):

FILE: tensorflow_estimator/python/estimator/canned/v1/dnn_testing_utils_v1.py
  function assert_close (line 66) | def assert_close(expected, actual, rtol=1e-04, message='', name='assert_...
  function create_checkpoint (line 82) | def create_checkpoint(weights_and_biases,
  function mock_head (line 130) | def mock_head(testcase, hidden_units, logits_dimension, expected_logits):
  function mock_optimizer (line 190) | def mock_optimizer(testcase, hidden_units, expected_loss=None):
  class BaseDNNModelFnTest (line 240) | class BaseDNNModelFnTest(object):
    method __init__ (line 243) | def __init__(self, dnn_model_fn, fc_impl=feature_column):
    method setUp (line 247) | def setUp(self):
    method tearDown (line 250) | def tearDown(self):
    method _test_logits (line 255) | def _test_logits(self, mode, hidden_units, logits_dimension, inputs,
    method test_one_dim_logits (line 287) | def test_one_dim_logits(self):
    method test_multi_dim_logits (line 311) | def test_multi_dim_logits(self):
    method test_multi_example_multi_dim_logits (line 336) | def test_multi_example_multi_dim_logits(self):
    method test_multi_dim_input_one_dim_logits (line 365) | def test_multi_dim_input_one_dim_logits(self):
    method test_multi_dim_input_multi_dim_logits (line 390) | def test_multi_dim_input_multi_dim_logits(self):
    method test_multi_feature_column_multi_dim_logits (line 415) | def test_multi_feature_column_multi_dim_logits(self):
    method test_multi_feature_column_mix_multi_dim_logits (line 466) | def test_multi_feature_column_mix_multi_dim_logits(self):
    method test_features_tensor_raises_value_error (line 517) | def test_features_tensor_raises_value_error(self):
  class BaseDNNLogitFnTest (line 545) | class BaseDNNLogitFnTest(object):
    method __init__ (line 548) | def __init__(self, dnn_logit_fn_builder, fc_impl=feature_column):
    method setUp (line 552) | def setUp(self):
    method tearDown (line 555) | def tearDown(self):
    method _test_logits (line 560) | def _test_logits(self,
    method test_one_dim_logits (line 594) | def test_one_dim_logits(self):
    method test_one_dim_logits_with_batch_norm (line 617) | def test_one_dim_logits_with_batch_norm(self):
    method test_multi_dim_logits (line 682) | def test_multi_dim_logits(self):
    method test_multi_example_multi_dim_logits (line 706) | def test_multi_example_multi_dim_logits(self):
    method test_multi_dim_input_one_dim_logits (line 734) | def test_multi_dim_input_one_dim_logits(self):
    method test_multi_dim_input_multi_dim_logits (line 759) | def test_multi_dim_input_multi_dim_logits(self):
    method test_multi_feature_column_multi_dim_logits (line 783) | def test_multi_feature_column_multi_dim_logits(self):
    method test_multi_feature_column_mix_multi_dim_logits (line 834) | def test_multi_feature_column_mix_multi_dim_logits(self):
  class BaseDNNWarmStartingTest (line 886) | class BaseDNNWarmStartingTest(object):
    method __init__ (line 888) | def __init__(self,
    method setUp (line 896) | def setUp(self):
    method tearDown (line 911) | def tearDown(self):
    method assertAllNotClose (line 916) | def assertAllNotClose(self, t1, t2):
    method test_classifier_basic_warm_starting (line 927) | def test_classifier_basic_warm_starting(self):
    method test_regressor_basic_warm_starting (line 960) | def test_regressor_basic_warm_starting(self):
    method test_warm_starting_selective_variables (line 991) | def test_warm_starting_selective_variables(self):
    method test_warm_starting_with_vocab_remapping_and_partitioning (line 1041) | def test_warm_starting_with_vocab_remapping_and_partitioning(self):
    method test_warm_starting_with_naming_change (line 1139) | def test_warm_starting_with_naming_change(self):
  class BaseDNNClassifierEvaluateTest (line 1190) | class BaseDNNClassifierEvaluateTest(object):
    method __init__ (line 1192) | def __init__(self, dnn_classifier_fn, fc_impl=feature_column):
    method setUp (line 1196) | def setUp(self):
    method tearDown (line 1199) | def tearDown(self):
    method test_one_dim (line 1204) | def test_one_dim(self):
    method test_multi_dim (line 1246) | def test_multi_dim(self):
    method test_float_labels (line 1283) | def test_float_labels(self):
    method test_multi_dim_weights (line 1310) | def test_multi_dim_weights(self):
  class BaseDNNRegressorEvaluateTest (line 1341) | class BaseDNNRegressorEvaluateTest(object):
    method __init__ (line 1343) | def __init__(self, dnn_regressor_fn, fc_impl=feature_column):
    method setUp (line 1347) | def setUp(self):
    method tearDown (line 1350) | def tearDown(self):
    method test_one_dim (line 1355) | def test_one_dim(self):
    method test_multi_dim (line 1387) | def test_multi_dim(self):
    method test_multi_dim_weights (line 1422) | def test_multi_dim_weights(self):
  class BaseDNNClassifierPredictTest (line 1452) | class BaseDNNClassifierPredictTest(object):
    method __init__ (line 1454) | def __init__(self, dnn_classifier_fn, fc_impl=feature_column):
    method setUp (line 1458) | def setUp(self):
    method tearDown (line 1461) | def tearDown(self):
    method _test_one_dim (line 1466) | def _test_one_dim(self, label_vocabulary, label_output_fn):
    method test_one_dim_without_label_vocabulary (line 1502) | def test_one_dim_without_label_vocabulary(self):
    method test_one_dim_with_label_vocabulary (line 1506) | def test_one_dim_with_label_vocabulary(self):
    method _test_multi_dim_with_3_classes (line 1512) | def _test_multi_dim_with_3_classes(self, label_vocabulary, label_outpu...
    method test_multi_dim_with_3_classes_but_no_label_vocab (line 1559) | def test_multi_dim_with_3_classes_but_no_label_vocab(self):
    method test_multi_dim_with_3_classes_and_label_vocab (line 1563) | def test_multi_dim_with_3_classes_and_label_vocab(self):
  class BaseDNNRegressorPredictTest (line 1570) | class BaseDNNRegressorPredictTest(object):
    method __init__ (line 1572) | def __init__(self, dnn_regressor_fn, fc_impl=feature_column):
    method setUp (line 1576) | def setUp(self):
    method tearDown (line 1579) | def tearDown(self):
    method test_one_dim (line 1584) | def test_one_dim(self):
    method test_multi_dim (line 1608) | def test_multi_dim(self):
  class _SummaryHook (line 1637) | class _SummaryHook(tf.compat.v1.train.SessionRunHook):
    method __init__ (line 1640) | def __init__(self):
    method begin (line 1643) | def begin(self):
    method before_run (line 1646) | def before_run(self, run_context):
    method after_run (line 1649) | def after_run(self, run_context, run_values):
    method summaries (line 1654) | def summaries(self):
  function _assert_checkpoint (line 1658) | def _assert_checkpoint(testcase, global_step, input_units, hidden_units,
  function _assert_simple_summary (line 1694) | def _assert_simple_summary(testcase, expected_values, actual_summary):
  class BaseDNNClassifierTrainTest (line 1710) | class BaseDNNClassifierTrainTest(object):
    method __init__ (line 1712) | def __init__(self, dnn_classifier_fn, fc_impl=feature_column):
    method setUp (line 1716) | def setUp(self):
    method tearDown (line 1719) | def tearDown(self):
    method test_from_scratch_with_default_optimizer_binary (line 1724) | def test_from_scratch_with_default_optimizer_binary(self):
    method test_from_scratch_with_default_optimizer_multi_class (line 1745) | def test_from_scratch_with_default_optimizer_multi_class(self):
    method test_from_scratch_validate_summary (line 1768) | def test_from_scratch_validate_summary(self):
    method test_binary_classification (line 1803) | def test_binary_classification(self):
    method test_binary_classification_float_labels (line 1856) | def test_binary_classification_float_labels(self):
    method test_multi_class (line 1888) | def test_multi_class(self):
  class BaseDNNRegressorTrainTest (line 1944) | class BaseDNNRegressorTrainTest(object):
    method __init__ (line 1946) | def __init__(self, dnn_regressor_fn, fc_impl=feature_column):
    method setUp (line 1950) | def setUp(self):
    method tearDown (line 1953) | def tearDown(self):
    method test_from_scratch_with_default_optimizer (line 1958) | def test_from_scratch_with_default_optimizer(self):
    method test_from_scratch (line 1979) | def test_from_scratch(self):
    method test_one_dim (line 2014) | def test_one_dim(self):
    method test_multi_dim (line 2068) | def test_multi_dim(self):

FILE: tensorflow_estimator/python/estimator/canned/v1/linear_estimator_test_v1.py
  function _linear_estimator_fn (line 36) | def _linear_estimator_fn(weight_column=None, label_dimension=1, **kwargs):
  class LinearEstimatorEvaluateTest (line 48) | class LinearEstimatorEvaluateTest(
    method __init__ (line 52) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearEstimatorPredictTest (line 59) | class LinearEstimatorPredictTest(
    method __init__ (line 62) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearEstimatorTrainTest (line 69) | class LinearEstimatorTrainTest(
    method __init__ (line 72) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearEstimatorIntegrationTest (line 79) | class LinearEstimatorIntegrationTest(tf.test.TestCase):
    method setUp (line 81) | def setUp(self):
    method tearDown (line 84) | def tearDown(self):
    method _test_complete_flow (line 89) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method test_numpy_input_fn (line 124) | def test_numpy_input_fn(self):

FILE: tensorflow_estimator/python/estimator/canned/v1/linear_test_v1.py
  function _linear_regressor_fn (line 31) | def _linear_regressor_fn(*args, **kwargs):
  function _linear_classifier_fn (line 35) | def _linear_classifier_fn(*args, **kwargs):
  class LinearRegressorPartitionerTest (line 43) | class LinearRegressorPartitionerTest(
    method __init__ (line 47) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearRegressorPartitionerV2Test (line 54) | class LinearRegressorPartitionerV2Test(
    method __init__ (line 58) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearRegressorEvaluationTest (line 65) | class LinearRegressorEvaluationTest(
    method __init__ (line 69) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearRegressorEvaluationV2Test (line 76) | class LinearRegressorEvaluationV2Test(
    method __init__ (line 80) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearRegressorPredictTest (line 87) | class LinearRegressorPredictTest(
    method __init__ (line 90) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearRegressorPredictV2Test (line 97) | class LinearRegressorPredictV2Test(
    method __init__ (line 100) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearRegressorIntegrationTest (line 107) | class LinearRegressorIntegrationTest(
    method __init__ (line 111) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearRegressorIntegrationV2Test (line 118) | class LinearRegressorIntegrationV2Test(
    method __init__ (line 122) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearRegressorTrainingTest (line 129) | class LinearRegressorTrainingTest(
    method __init__ (line 132) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearRegressorTrainingV2Test (line 139) | class LinearRegressorTrainingV2Test(
    method __init__ (line 142) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearClassifierTrainingTest (line 150) | class LinearClassifierTrainingTest(
    method __init__ (line 153) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearClassifierTrainingV2Test (line 160) | class LinearClassifierTrainingV2Test(
    method __init__ (line 163) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearClassifierEvaluationTest (line 172) | class LinearClassifierEvaluationTest(
    method __init__ (line 176) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearClassifierEvaluationV2Test (line 183) | class LinearClassifierEvaluationV2Test(
    method __init__ (line 187) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearClassifierPredictTest (line 196) | class LinearClassifierPredictTest(
    method __init__ (line 199) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearClassifierPredictV2Test (line 206) | class LinearClassifierPredictV2Test(
    method __init__ (line 209) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearClassifierIntegrationTest (line 218) | class LinearClassifierIntegrationTest(
    method __init__ (line 222) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearClassifierIntegrationV2Test (line 229) | class LinearClassifierIntegrationV2Test(
    method __init__ (line 233) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearLogitFnTest (line 243) | class LinearLogitFnTest(linear_testing_utils_v1.BaseLinearLogitFnTest,
    method __init__ (line 246) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearLogitFnV2Test (line 253) | class LinearLogitFnV2Test(linear_testing_utils_v1.BaseLinearLogitFnTest,
    method __init__ (line 256) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearWarmStartingTest (line 264) | class LinearWarmStartingTest(linear_testing_utils_v1.BaseLinearWarmStart...
    method __init__ (line 267) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearWarmStartingV2Test (line 277) | class LinearWarmStartingV2Test(
    method __init__ (line 280) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class ComputeFractionOfZeroTest (line 290) | class ComputeFractionOfZeroTest(tf.test.TestCase):
    method _assertSparsity (line 292) | def _assertSparsity(self, expected_sparsity, tensor):
    method test_small_float32 (line 297) | def test_small_float32(self):
    method test_small_int32 (line 303) | def test_small_int32(self):
    method test_small_float64 (line 307) | def test_small_float64(self):
    method test_small_int64 (line 311) | def test_small_int64(self):
    method test_nested (line 315) | def test_nested(self):
    method test_none (line 320) | def test_none(self):
    method test_empty (line 324) | def test_empty(self):
    method test_multiple_empty (line 331) | def test_multiple_empty(self):
    method test_some_empty (line 341) | def test_some_empty(self):
    method test_mixed_types (line 349) | def test_mixed_types(self):
    method test_2_27_zeros__using_512_MiB_of_ram (line 356) | def test_2_27_zeros__using_512_MiB_of_ram(self):
    method test_2_27_ones__using_512_MiB_of_ram (line 360) | def test_2_27_ones__using_512_MiB_of_ram(self):

FILE: tensorflow_estimator/python/estimator/canned/v1/linear_testing_utils_v1.py
  function assert_close (line 72) | def assert_close(expected, actual, rtol=1e-04, name='assert_close'):
  function save_variables_to_ckpt (line 87) | def save_variables_to_ckpt(model_dir):
  function queue_parsed_features (line 94) | def queue_parsed_features(feature_map):
  function sorted_key_dict (line 109) | def sorted_key_dict(unsorted_dict):
  function sigmoid (line 113) | def sigmoid(x):
  class CheckPartitionerVarHook (line 117) | class CheckPartitionerVarHook(tf.compat.v1.train.SessionRunHook):
    method __init__ (line 120) | def __init__(self, test_case, var_name, var_dim, partitions):
    method begin (line 126) | def begin(self):
  class BaseLinearRegressorPartitionerTest (line 139) | class BaseLinearRegressorPartitionerTest(object):
    method __init__ (line 141) | def __init__(self, linear_regressor_fn, fc_lib=feature_column):
    method setUp (line 145) | def setUp(self):
    method tearDown (line 148) | def tearDown(self):
    method testPartitioner (line 153) | def testPartitioner(self):
    method testDefaultPartitionerWithMultiplePsReplicas (line 181) | def testDefaultPartitionerWithMultiplePsReplicas(self):
  class BaseLinearRegressorEvaluationTest (line 219) | class BaseLinearRegressorEvaluationTest(object):
    method __init__ (line 221) | def __init__(self, linear_regressor_fn, fc_lib=feature_column):
    method setUp (line 225) | def setUp(self):
    method tearDown (line 228) | def tearDown(self):
    method test_evaluation_for_simple_data (line 233) | def test_evaluation_for_simple_data(self):
    method test_evaluation_batch (line 259) | def test_evaluation_batch(self):
    method test_evaluation_weights (line 289) | def test_evaluation_weights(self):
    method test_evaluation_for_multi_dimensions (line 322) | def test_evaluation_for_multi_dimensions(self):
    method test_evaluation_for_multiple_feature_columns (line 358) | def test_evaluation_for_multiple_feature_columns(self):
    method test_evaluation_for_multiple_feature_columns_mix (line 396) | def test_evaluation_for_multiple_feature_columns_mix(self):
  class BaseLinearRegressorPredictTest (line 436) | class BaseLinearRegressorPredictTest(object):
    method __init__ (line 438) | def __init__(self, linear_regressor_fn, fc_lib=feature_column):
    method setUp (line 442) | def setUp(self):
    method tearDown (line 445) | def tearDown(self):
    method test_1d (line 450) | def test_1d(self):
    method testMultiDim (line 473) | def testMultiDim(self):
    method testTwoFeatureColumns (line 506) | def testTwoFeatureColumns(self):
    method testTwoFeatureColumnsMix (line 534) | def testTwoFeatureColumnsMix(self):
    method testSparseCombiner (line 559) | def testSparseCombiner(self):
  class BaseLinearRegressorIntegrationTest (line 622) | class BaseLinearRegressorIntegrationTest(object):
    method __init__ (line 624) | def __init__(self, linear_regressor_fn, fc_lib=feature_column):
    method setUp (line 628) | def setUp(self):
    method tearDown (line 631) | def tearDown(self):
    method _test_complete_flow (line 636) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method test_numpy_input_fn (line 669) | def test_numpy_input_fn(self):
    method test_pandas_input_fn (line 705) | def test_pandas_input_fn(self):
    method test_input_fn_from_parse_example (line 734) | def test_input_fn_from_parse_example(self):
  class BaseLinearRegressorTrainingTest (line 795) | class BaseLinearRegressorTrainingTest(object):
    method __init__ (line 797) | def __init__(self, linear_regressor_fn, fc_lib=feature_column):
    method setUp (line 801) | def setUp(self):
    method tearDown (line 804) | def tearDown(self):
    method _mock_optimizer (line 809) | def _mock_optimizer(self, expected_loss=None):
    method _assert_checkpoint (line 847) | def _assert_checkpoint(self,
    method testFromScratchWithDefaultOptimizer (line 872) | def testFromScratchWithDefaultOptimizer(self):
    method testTrainWithOneDimLabel (line 888) | def testTrainWithOneDimLabel(self):
    method testTrainWithOneDimWeight (line 908) | def testTrainWithOneDimWeight(self):
    method testFromScratch (line 933) | def testFromScratch(self):
    method testFromCheckpoint (line 957) | def testFromCheckpoint(self):
    method testFromCheckpointMultiBatch (line 992) | def testFromCheckpointMultiBatch(self):
  class BaseLinearClassifierTrainingTest (line 1031) | class BaseLinearClassifierTrainingTest(object):
    method __init__ (line 1033) | def __init__(self, linear_classifier_fn, fc_lib=feature_column):
    method setUp (line 1037) | def setUp(self):
    method tearDown (line 1040) | def tearDown(self):
    method _mock_optimizer (line 1044) | def _mock_optimizer(self, expected_loss=None):
    method _assert_checkpoint (line 1078) | def _assert_checkpoint(self,
    method _testFromScratchWithDefaultOptimizer (line 1107) | def _testFromScratchWithDefaultOptimizer(self, n_classes):
    method testBinaryClassesFromScratchWithDefaultOptimizer (line 1123) | def testBinaryClassesFromScratchWithDefaultOptimizer(self):
    method testMultiClassesFromScratchWithDefaultOptimizer (line 1126) | def testMultiClassesFromScratchWithDefaultOptimizer(self):
    method _testTrainWithTwoDimsLabel (line 1129) | def _testTrainWithTwoDimsLabel(self, n_classes):
    method testBinaryClassesTrainWithTwoDimsLabel (line 1150) | def testBinaryClassesTrainWithTwoDimsLabel(self):
    method testMultiClassesTrainWithTwoDimsLabel (line 1153) | def testMultiClassesTrainWithTwoDimsLabel(self):
    method _testTrainWithOneDimLabel (line 1156) | def _testTrainWithOneDimLabel(self, n_classes):
    method testBinaryClassesTrainWithOneDimLabel (line 1175) | def testBinaryClassesTrainWithOneDimLabel(self):
    method testMultiClassesTrainWithOneDimLabel (line 1178) | def testMultiClassesTrainWithOneDimLabel(self):
    method _testTrainWithTwoDimsWeight (line 1181) | def _testTrainWithTwoDimsWeight(self, n_classes):
    method testBinaryClassesTrainWithTwoDimsWeight (line 1206) | def testBinaryClassesTrainWithTwoDimsWeight(self):
    method testMultiClassesTrainWithTwoDimsWeight (line 1209) | def testMultiClassesTrainWithTwoDimsWeight(self):
    method _testTrainWithOneDimWeight (line 1212) | def _testTrainWithOneDimWeight(self, n_classes):
    method testBinaryClassesTrainWithOneDimWeight (line 1235) | def testBinaryClassesTrainWithOneDimWeight(self):
    method testMultiClassesTrainWithOneDimWeight (line 1238) | def testMultiClassesTrainWithOneDimWeight(self):
    method _testFromScratch (line 1241) | def _testFromScratch(self, n_classes):
    method testBinaryClassesFromScratch (line 1277) | def testBinaryClassesFromScratch(self):
    method testMultiClassesFromScratch (line 1280) | def testMultiClassesFromScratch(self):
    method _testFromCheckpoint (line 1283) | def _testFromCheckpoint(self, n_classes):
    method testBinaryClassesFromCheckpoint (line 1342) | def testBinaryClassesFromCheckpoint(self):
    method testMultiClassesFromCheckpoint (line 1345) | def testMultiClassesFromCheckpoint(self):
    method _testFromCheckpointFloatLabels (line 1348) | def _testFromCheckpointFloatLabels(self, n_classes):
    method testBinaryClassesFromCheckpointFloatLabels (line 1387) | def testBinaryClassesFromCheckpointFloatLabels(self):
    method testMultiClassesFromCheckpointFloatLabels (line 1390) | def testMultiClassesFromCheckpointFloatLabels(self):
    method _testFromCheckpointMultiBatch (line 1393) | def _testFromCheckpointMultiBatch(self, n_classes):
    method testBinaryClassesFromCheckpointMultiBatch (line 1457) | def testBinaryClassesFromCheckpointMultiBatch(self):
    method testMultiClassesFromCheckpointMultiBatch (line 1460) | def testMultiClassesFromCheckpointMultiBatch(self):
  class BaseLinearClassifierEvaluationTest (line 1464) | class BaseLinearClassifierEvaluationTest(object):
    method __init__ (line 1466) | def __init__(self, linear_classifier_fn, fc_lib=feature_column):
    method setUp (line 1470) | def setUp(self):
    method tearDown (line 1473) | def tearDown(self):
    method _test_evaluation_for_simple_data (line 1477) | def _test_evaluation_for_simple_data(self, n_classes):
    method test_binary_classes_evaluation_for_simple_data (line 1539) | def test_binary_classes_evaluation_for_simple_data(self):
    method test_multi_classes_evaluation_for_simple_data (line 1542) | def test_multi_classes_evaluation_for_simple_data(self):
    method _test_evaluation_batch (line 1545) | def _test_evaluation_batch(self, n_classes):
    method test_binary_classes_evaluation_batch (line 1617) | def test_binary_classes_evaluation_batch(self):
    method test_multi_classes_evaluation_batch (line 1620) | def test_multi_classes_evaluation_batch(self):
    method _test_evaluation_weights (line 1623) | def _test_evaluation_weights(self, n_classes):
    method test_binary_classes_evaluation_weights (line 1708) | def test_binary_classes_evaluation_weights(self):
    method test_multi_classes_evaluation_weights (line 1711) | def test_multi_classes_evaluation_weights(self):
  class BaseLinearClassifierPredictTest (line 1715) | class BaseLinearClassifierPredictTest(object):
    method __init__ (line 1717) | def __init__(self, linear_classifier_fn, fc_lib=feature_column):
    method setUp (line 1721) | def setUp(self):
    method tearDown (line 1724) | def tearDown(self):
    method _testPredictions (line 1728) | def _testPredictions(self, n_classes, label_vocabulary, label_output_fn):
    method testBinaryClassesWithoutLabelVocabulary (line 1803) | def testBinaryClassesWithoutLabelVocabulary(self):
    method testBinaryClassesWithLabelVocabulary (line 1810) | def testBinaryClassesWithLabelVocabulary(self):
    method testMultiClassesWithoutLabelVocabulary (line 1817) | def testMultiClassesWithoutLabelVocabulary(self):
    method testMultiClassesWithLabelVocabulary (line 1824) | def testMultiClassesWithLabelVocabulary(self):
    method testSparseCombiner (line 1831) | def testSparseCombiner(self):
  class BaseLinearClassifierIntegrationTest (line 1894) | class BaseLinearClassifierIntegrationTest(object):
    method __init__ (line 

[
  {
    "path": ".bazelrc",
    "chars": 130,
    "preview": "\n# Default options should come above this line\n\n# Put user-specific options in .bazelrc.user\ntry-import %workspace%/.baz"
  },
  {
    "path": ".gitignore",
    "chars": 135,
    "preview": "# editor files\n*.swp\n*~\n.vscode/\n.DS_Store\n\n# bazel\n/.bazelrc.user\n/bazel-*\n\n# python\n*.pyc\n*.pyo\n__pycache__\n*.whl\n.ipy"
  },
  {
    "path": "BUILD",
    "chars": 101,
    "preview": "# Description: Tensorflow Estimator.\n\nlicenses([\"notice\"])  # Apache 2.0\n\nexports_files([\"LICENSE\"])\n"
  },
  {
    "path": "CONTRIBUTING.md",
    "chars": 1453,
    "preview": "Want to contribute? Great! First, read this page (including the small print at the end).\n\n### Before you contribute\n\nBef"
  },
  {
    "path": "LICENSE",
    "chars": 11416,
    "preview": "Copyright 2018 The TensorFlow Authors.  All rights reserved.\n\n                                 Apache License\n          "
  },
  {
    "path": "README.md",
    "chars": 2010,
    "preview": "-----------------\n| **`Documentation`** |\n|-----------------|\n| [![Documentation](https://img.shields.io/badge/api-refer"
  },
  {
    "path": "WORKSPACE",
    "chars": 470,
    "preview": "workspace(name = \"org_tensorflow_estimator\")\n\n# Use a custom python toolchain to make sure we always use the python bina"
  },
  {
    "path": "tensorflow_estimator/BUILD",
    "chars": 2914,
    "preview": "# Placeholder: load py_library\n\n# Description: Tensorflow Estimator.\nload(\n    \"//tensorflow_estimator/python/estimator/"
  },
  {
    "path": "tensorflow_estimator/estimator.bzl",
    "chars": 736,
    "preview": "\"\"\"Estimator common skylark macros.\"\"\"\n\n# Macro to run Estimator py_tests against pip installation.\ndef py_test(deps = ["
  },
  {
    "path": "tensorflow_estimator/python/estimator/BUILD",
    "chars": 66530,
    "preview": "# Placeholder: load py_library\nload(\"//tensorflow_estimator:estimator.bzl\", \"py_test\")\n\npackage(default_visibility = [\"/"
  },
  {
    "path": "tensorflow_estimator/python/estimator/api/BUILD",
    "chars": 1933,
    "preview": "# Placeholder: load aliased py_binary\nload(\"//tensorflow_estimator/python/estimator/api:api_gen.bzl\", \"ESTIMATOR_API_INI"
  },
  {
    "path": "tensorflow_estimator/python/estimator/api/api_gen.bzl",
    "chars": 13914,
    "preview": "\"\"\"Targets for generating TensorFlow Estimator Python API __init__.py files.\n\nThis bzl file is copied with slight modifi"
  },
  {
    "path": "tensorflow_estimator/python/estimator/api/extractor_wrapper.py",
    "chars": 898,
    "preview": "# Copyright 2023 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/api/generator_wrapper.py",
    "chars": 897,
    "preview": "# Copyright 2023 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/__init__.py",
    "chars": 0,
    "preview": ""
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/baseline.py",
    "chars": 23473,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/baseline_estimator_test.py",
    "chars": 14944,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/baseline_test.py",
    "chars": 54404,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/canned_estimator_ds_integration_test.py",
    "chars": 5863,
    "preview": "# Copyright 2019 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/dnn.py",
    "chars": 48618,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/dnn_estimator_test.py",
    "chars": 7401,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/dnn_linear_combined.py",
    "chars": 48471,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/dnn_linear_combined_estimator_test.py",
    "chars": 9505,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/dnn_linear_combined_test.py",
    "chars": 38700,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/dnn_test_fc_v2.py",
    "chars": 19108,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/dnn_testing_utils.py",
    "chars": 82124,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/head.py",
    "chars": 72566,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/head_test.py",
    "chars": 166796,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/kmeans.py",
    "chars": 20395,
    "preview": "# Copyright 2016 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/kmeans_test.py",
    "chars": 21558,
    "preview": "# Copyright 2016 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/linear.py",
    "chars": 68128,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/linear_estimator_test.py",
    "chars": 7183,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/linear_model_test.py",
    "chars": 59598,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/linear_optimizer/BUILD",
    "chars": 2480,
    "preview": "# Placeholder: load py_library\nload(\"//tensorflow_estimator:estimator.bzl\", \"py_test\")\n\npackage(default_visibility = [\"/"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/linear_optimizer/__init__.py",
    "chars": 994,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/linear_optimizer/doc/sdca.ipynb",
    "chars": 11291,
    "preview": "{\n  \"cells\": [\n    {\n      \"cell_type\": \"markdown\",\n      \"metadata\": {\n        \"colab_type\": \"text\",\n        \"id\": \"DzJ"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/linear_optimizer/python/sdca_test.py",
    "chars": 26650,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/linear_optimizer/python/utils/sdca_ops.py",
    "chars": 32502,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/linear_optimizer/python/utils/sdca_ops_test.py",
    "chars": 54340,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/linear_optimizer/python/utils/sharded_mutable_dense_hashtable.py",
    "chars": 14506,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/linear_optimizer/python/utils/sharded_mutable_dense_hashtable_test.py",
    "chars": 4582,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/linear_test.py",
    "chars": 7789,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/linear_testing_utils.py",
    "chars": 84267,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/metric_keys.py",
    "chars": 2364,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/optimizers.py",
    "chars": 6396,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/optimizers_test.py",
    "chars": 4324,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/optimizers_test_v2.py",
    "chars": 5806,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/parsing_utils.py",
    "chars": 15354,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/parsing_utils_test.py",
    "chars": 10653,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/prediction_keys.py",
    "chars": 1275,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/rnn.py",
    "chars": 28806,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/rnn_test.py",
    "chars": 67981,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/saved_model_estimator.py",
    "chars": 20175,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/saved_model_estimator_test.py",
    "chars": 20627,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/testdata/wire_vocabulary.txt",
    "chars": 20,
    "preview": "omar\nstringer\nmarlo\n"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/timeseries/BUILD",
    "chars": 5297,
    "preview": "# Placeholder: load py_library\nload(\"//tensorflow_estimator:estimator.bzl\", \"py_test\")\n\npackage(default_visibility = [\"/"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/timeseries/ar_model.py",
    "chars": 38400,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/timeseries/ar_model_test.py",
    "chars": 5182,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/timeseries/ar_model_training_test.py",
    "chars": 8601,
    "preview": "# Copyright 2019 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/timeseries/estimators.py",
    "chars": 20770,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/timeseries/estimators_test.py",
    "chars": 8722,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/timeseries/feature_keys.py",
    "chars": 2314,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/timeseries/head.py",
    "chars": 21379,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/timeseries/head_test.py",
    "chars": 17922,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/timeseries/math_utils.py",
    "chars": 19761,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/timeseries/math_utils_test.py",
    "chars": 4079,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/timeseries/model.py",
    "chars": 14441,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/timeseries/model_utils.py",
    "chars": 3212,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/timeseries/saved_model_utils.py",
    "chars": 14941,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/timeseries/state_management.py",
    "chars": 4004,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/v1/__init__.py",
    "chars": 0,
    "preview": ""
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/v1/baseline_estimator_test_v1.py",
    "chars": 15279,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/v1/baseline_test_v1.py",
    "chars": 55981,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/v1/dnn_estimator_test_v1.py",
    "chars": 6245,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/v1/dnn_linear_combined_estimator_test_v1.py",
    "chars": 8836,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/v1/dnn_linear_combined_test_v1.py",
    "chars": 46937,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/v1/dnn_test_fc_v1_v1.py",
    "chars": 17817,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/v1/dnn_test_fc_v2_v1.py",
    "chars": 17722,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/v1/dnn_testing_utils_v1.py",
    "chars": 82598,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/v1/linear_estimator_test_v1.py",
    "chars": 5629,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/v1/linear_test_v1.py",
    "chars": 13879,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/canned/v1/linear_testing_utils_v1.py",
    "chars": 91091,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/distribute_strategy_estimator_integration_test.py",
    "chars": 7834,
    "preview": "# Copyright 2016 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/distribute_strategy_estimator_training_test.py",
    "chars": 24488,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/early_stopping.py",
    "chars": 23610,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/early_stopping_test.py",
    "chars": 9670,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/estimator.py",
    "chars": 103213,
    "preview": "# Copyright 2016 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/estimator_export.py",
    "chars": 2883,
    "preview": "# Copyright 2023 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/estimator_export_test.py",
    "chars": 2004,
    "preview": "# Copyright 2023 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/estimator_lib.py",
    "chars": 4886,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/estimator_test.py",
    "chars": 134806,
    "preview": "# Copyright 2016 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/export/__init__.py",
    "chars": 0,
    "preview": ""
  },
  {
    "path": "tensorflow_estimator/python/estimator/export/export.py",
    "chars": 20652,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/export/export_lib.py",
    "chars": 3055,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/export/export_output.py",
    "chars": 1924,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/export/export_test.py",
    "chars": 23540,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/export/function.py",
    "chars": 13836,
    "preview": "# Copyright 2019 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/export/function_test.py",
    "chars": 9491,
    "preview": "# Copyright 2019 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/exporter.py",
    "chars": 19873,
    "preview": "# Copyright 2016 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/exporter_test.py",
    "chars": 17430,
    "preview": "# Copyright 2016 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/extenders.py",
    "chars": 4896,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/extenders_test.py",
    "chars": 5205,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/gc.py",
    "chars": 6327,
    "preview": "# Copyright 2016 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/gc_test.py",
    "chars": 6074,
    "preview": "# Copyright 2016 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/head/__init__.py",
    "chars": 0,
    "preview": ""
  },
  {
    "path": "tensorflow_estimator/python/estimator/head/base_head.py",
    "chars": 39719,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/head/base_head_test.py",
    "chars": 10317,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/head/binary_class_head.py",
    "chars": 27000,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/head/binary_class_head_test.py",
    "chars": 64901,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/head/head_utils.py",
    "chars": 3986,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/head/multi_class_head.py",
    "chars": 21524,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/head/multi_class_head_test.py",
    "chars": 65188,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/head/multi_head.py",
    "chars": 23371,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/head/multi_head_test.py",
    "chars": 36820,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/head/multi_label_head.py",
    "chars": 26898,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/head/multi_label_head_test.py",
    "chars": 56975,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/head/regression_head.py",
    "chars": 23862,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/head/regression_head_test.py",
    "chars": 72196,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/head/sequential_head.py",
    "chars": 20736,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/head/sequential_head_test.py",
    "chars": 21047,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/hooks/__init__.py",
    "chars": 0,
    "preview": ""
  },
  {
    "path": "tensorflow_estimator/python/estimator/hooks/basic_session_run_hooks.py",
    "chars": 2835,
    "preview": "# Copyright 2016 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/hooks/basic_session_run_hooks_test.py",
    "chars": 58286,
    "preview": "# pylint: disable=g-bad-file-header\n# Copyright 2016 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the"
  },
  {
    "path": "tensorflow_estimator/python/estimator/hooks/fake_summary_writer.py",
    "chars": 5595,
    "preview": "# Copyright 2015 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/hooks/hooks.py",
    "chars": 11072,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/hooks/hooks_test.py",
    "chars": 15092,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/hooks/session_run_hook.py",
    "chars": 4204,
    "preview": "# Copyright 2016 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/inputs/__init__.py",
    "chars": 0,
    "preview": ""
  },
  {
    "path": "tensorflow_estimator/python/estimator/inputs/inputs.py",
    "chars": 1106,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/inputs/numpy_io.py",
    "chars": 8018,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/inputs/numpy_io_test.py",
    "chars": 22399,
    "preview": "# Copyright 2016 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/inputs/pandas_io.py",
    "chars": 5859,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/inputs/pandas_io_test.py",
    "chars": 11028,
    "preview": "# Copyright 2015 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/inputs/queues/__init__.py",
    "chars": 0,
    "preview": ""
  },
  {
    "path": "tensorflow_estimator/python/estimator/inputs/queues/feeding_functions.py",
    "chars": 18804,
    "preview": "# Copyright 2016 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/inputs/queues/feeding_functions_test.py",
    "chars": 13543,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/inputs/queues/feeding_queue_runner.py",
    "chars": 6883,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/inputs/queues/feeding_queue_runner_test.py",
    "chars": 5162,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/keras_distribute_strategy_test.py",
    "chars": 11635,
    "preview": "# Copyright 2016 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/keras_lib.py",
    "chars": 33679,
    "preview": "# Copyright 2015 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/keras_premade_model_test.py",
    "chars": 8192,
    "preview": "# Copyright 2019 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/keras_test.py",
    "chars": 48113,
    "preview": "# Copyright 2016 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/mode_keys.py",
    "chars": 1080,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/model_fn.py",
    "chars": 25073,
    "preview": "# Copyright 2016 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/model_fn_test.py",
    "chars": 26991,
    "preview": "# Copyright 2016 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/object_checkpointing_test.py",
    "chars": 5767,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/run_config.py",
    "chars": 38460,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/run_config_test.py",
    "chars": 46693,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/tf_estimator_doctest.py",
    "chars": 5800,
    "preview": "# Copyright 2019 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/tools/__init__.py",
    "chars": 0,
    "preview": ""
  },
  {
    "path": "tensorflow_estimator/python/estimator/tools/analytics.py",
    "chars": 1265,
    "preview": "# Copyright 2019 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/tools/checkpoint_converter.py",
    "chars": 14988,
    "preview": "# Copyright 2019 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/tools/checkpoint_converter_test.py",
    "chars": 16018,
    "preview": "# Copyright 2019 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/tpu/BUILD",
    "chars": 6237,
    "preview": "# Description: TPUEstimator\n\n# Placeholder: load py_library\n\n# INTERNAL TEST RULE PLACEHOLDER\nload(\"//tensorflow_estimat"
  },
  {
    "path": "tensorflow_estimator/python/estimator/tpu/__init__.py",
    "chars": 818,
    "preview": "# Copyright 2019 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/tpu/_tpu_estimator_embedding.py",
    "chars": 28939,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/tpu/autotuning_iterations_per_loop_test.py",
    "chars": 18228,
    "preview": "# Copyright 2019 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/tpu/error_handling.py",
    "chars": 5327,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/tpu/error_handling_test.py",
    "chars": 1481,
    "preview": "# Copyright 2019 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/tpu/iteration_count_estimator.py",
    "chars": 7833,
    "preview": "# Copyright 2019 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/tpu/spatial_partitioning_api.md",
    "chars": 3578,
    "preview": "# Spatial partitioning\n\nSpatial partitioning allows us to run models with larger input images. Typically\nthese models wi"
  },
  {
    "path": "tensorflow_estimator/python/estimator/tpu/tpu_config.py",
    "chars": 15727,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/tpu/tpu_config_test.py",
    "chars": 8067,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/tpu/tpu_context.py",
    "chars": 34455,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/tpu/tpu_enqueue_sequence_test.py",
    "chars": 17168,
    "preview": "# Copyright 2021 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/tpu/tpu_estimator.py",
    "chars": 184949,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/tpu/tpu_estimator_embedding_test.py",
    "chars": 57098,
    "preview": "# Copyright 2021 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/tpu/tpu_estimator_evaluation_test.py",
    "chars": 15747,
    "preview": "# Copyright 2021 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/tpu/tpu_estimator_export_test.py",
    "chars": 28157,
    "preview": "# Copyright 2021 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/tpu/tpu_estimator_gradients_test.py",
    "chars": 32083,
    "preview": "# Copyright 2021 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/tpu/tpu_estimator_input_v2_test.py",
    "chars": 6856,
    "preview": "# Copyright 2021 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/tpu/tpu_estimator_integration_test.py",
    "chars": 21802,
    "preview": "# Copyright 2021 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/tpu/tpu_estimator_model_parallelism_test.py",
    "chars": 13734,
    "preview": "# Copyright 2021 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/tpu/tpu_estimator_signals_test.py",
    "chars": 13742,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/tpu/tpu_estimator_test.py",
    "chars": 104675,
    "preview": "# Copyright 2020 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/tpu/util.py",
    "chars": 3607,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/training.py",
    "chars": 44016,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/training_test.py",
    "chars": 90338,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/util.py",
    "chars": 4635,
    "preview": "# Copyright 2017 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/python/estimator/util_test.py",
    "chars": 4017,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/tools/pip_package/BUILD",
    "chars": 873,
    "preview": "package(default_visibility = [\"//tensorflow_estimator:internal\"])\n\n# Description:\n#  Tools for building the TensorFlow p"
  },
  {
    "path": "tensorflow_estimator/tools/pip_package/build_pip_package.sh",
    "chars": 4038,
    "preview": "#!/usr/bin/env bash\n# Copyright 2015 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License,"
  },
  {
    "path": "tensorflow_estimator/tools/pip_package/create_pip_helper.py",
    "chars": 5275,
    "preview": "# Copyright 2019 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "tensorflow_estimator/tools/pip_package/setup.py",
    "chars": 2968,
    "preview": "# Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n#\n# Licensed under the Apache License, Version 2.0 (the \"L"
  },
  {
    "path": "third_party/py/BUILD",
    "chars": 0,
    "preview": ""
  },
  {
    "path": "third_party/py/BUILD.tpl",
    "chars": 766,
    "preview": "licenses([\"restricted\"])\n\npackage(default_visibility = [\"//visibility:public\"])\n\n# Point both runtimes to the same pytho"
  },
  {
    "path": "third_party/py/python_configure.bzl",
    "chars": 2144,
    "preview": "\"\"\"Repository rule for Python autoconfiguration.\n\n`python_configure` depends on the following environment variables:\n\n  "
  }
]