diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-06-20 08:55:35 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-06-20 08:59:31 -0700 |
| commit | f580c21053c47a12b3b816cc50c261479f36db5d (patch) | |
| tree | 839ca8cb1f8128d064e958e04251fd4e5b718fcf /tensorflow/contrib/tensor_forest/python | |
| parent | 95d91a3e065d7ddd3a0201af61e5e78330449d37 (diff) | |
Fix TensorForest feature bagging when using feature-column data.
PiperOrigin-RevId: 159566463
Diffstat (limited to 'tensorflow/contrib/tensor_forest/python')
| -rw-r--r-- | tensorflow/contrib/tensor_forest/python/tensor_forest.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/tensor_forest/python/tensor_forest.py b/tensorflow/contrib/tensor_forest/python/tensor_forest.py index 177783c207..bdfe981baf 100644 --- a/tensorflow/contrib/tensor_forest/python/tensor_forest.py +++ b/tensorflow/contrib/tensor_forest/python/tensor_forest.py @@ -435,7 +435,7 @@ class RandomForestGraphs(object): if processed_sparse_features is not None: raise NotImplementedError( 'Feature bagging not supported with sparse features.') - tree_data = self._bag_features(i, input_data) + tree_data = self._bag_features(i, tree_data) probabilities.append(self.trees[i].inference_graph( tree_data, data_spec, |