/estimator

TensorFlow Estimator

Primary LanguagePythonApache License 2.0Apache-2.0


Documentation
Documentation

TensorFlow Estimator is a high-level TensorFlow API that greatly simplifies machine learning programming. Estimators encapsulate training, evaluation, prediction, and exporting for your model.

Getting Started

See our Estimator getting started guide for an introduction to the Estimator APIs.

Installation

tf.estimator is installed when you install the TensorFlow pip package. See Installing TensorFlow for instructions.

Developing

If you want to build TensorFlow Estimator locally, you will need to install Bazel and install TensorFlow.

# To build TensorFlow Estimator whl file.
bazel build //tensorflow_estimator/tools/pip_package:build_pip_package
bazel-bin/tensorflow_estimator/tools/pip_package/build_pip_package /tmp/estimator_pip

# To run all Estimator tests
bazel test //tensorflow_estimator/...

Contribution guidelines

If you want to contribute to TensorFlow Estimator, be sure to review the contribution guidelines.

Note that this repository is included as a component of the main TensorFlow package, and any issues encountered while using Estimators should be filed under TensorFlow GitHub Issues, as we do not separately track issues in this repository. You can link this repository in any issues created as necessary.

Please see TensorFlow Discuss for general questions and discussion and please direct specific questions to Stack Overflow.

License

Apache License 2.0