diff options
author | 2016-06-30 14:40:02 -0800 | |
---|---|---|
committer | 2016-06-30 15:48:03 -0700 | |
commit | b70103502b41df370906e8988b6593e55caf69cf (patch) | |
tree | 3455ed439430bb6c0e739bb974a52a99a7bc6626 /tensorflow/contrib/tensor_forest/__init__.py | |
parent | d3067c338425bdf97fa782d834399b89bce18309 (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__.py | 2 |
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 * |