diff options
author | Younghee Kwon <youngheek@google.com> | 2018-03-29 09:43:19 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-03-29 09:48:47 -0700 |
commit | a2b6c3c124664d682094a1ecfa9cc00cca8ada85 (patch) | |
tree | 1f72bb1a6e6542bfd5e215b535e46b9e782459d5 /tensorflow/python/__init__.py | |
parent | ae3d20f9aef78554f0d0f5eec13982e9802a45d2 (diff) |
Added kernels and estimators for Gradient Boosting Trees algorithm.
BoostedTreesClassifier and BoostedTreesRegressor are added to tf.estimator.
Also some training utility functions are added to tf.contrib.estimator.
PiperOrigin-RevId: 190942599
Diffstat (limited to 'tensorflow/python/__init__.py')
-rw-r--r-- | tensorflow/python/__init__.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/python/__init__.py b/tensorflow/python/__init__.py index 3346937904..ab1d01a835 100644 --- a/tensorflow/python/__init__.py +++ b/tensorflow/python/__init__.py @@ -98,6 +98,8 @@ from tensorflow.python.summary import summary from tensorflow.python.user_ops import user_ops from tensorflow.python.util import compat +# Import boosted trees ops to make sure the ops are registered (but unused). +from tensorflow.python.ops import gen_boosted_trees_ops as _gen_boosted_trees_ops # Import cudnn rnn ops to make sure their ops are registered. from tensorflow.python.ops import gen_cudnn_rnn_ops as _ |