tensorflow / estimator Goto Github PK

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.

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

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

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/...

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.

Apache License 2.0