aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/tensor_forest/__init__.py
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2016-06-30 14:40:02 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2016-06-30 15:48:03 -0700
commitb70103502b41df370906e8988b6593e55caf69cf (patch)
tree3455ed439430bb6c0e739bb974a52a99a7bc6626 /tensorflow/contrib/tensor_forest/__init__.py
parentd3067c338425bdf97fa782d834399b89bce18309 (diff)
Improvements to tensor_forest, including support for sparse and categorical inputs.
Add tf.learn.Estimator for random forests. Change: 126352221
Diffstat (limited to 'tensorflow/contrib/tensor_forest/__init__.py')
-rw-r--r--tensorflow/contrib/tensor_forest/__init__.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/contrib/tensor_forest/__init__.py b/tensorflow/contrib/tensor_forest/__init__.py
index 7cf05299c4..7d97e01df0 100644
--- a/tensorflow/contrib/tensor_forest/__init__.py
+++ b/tensorflow/contrib/tensor_forest/__init__.py
@@ -18,4 +18,6 @@ from __future__ import division
from __future__ import print_function
# pylint: disable=unused-import,wildcard-import
+from tensorflow.contrib.tensor_forest.client import *
+from tensorflow.contrib.tensor_forest.data import *
from tensorflow.contrib.tensor_forest.python import *