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