diff options
author | 2017-10-02 13:30:00 -0700 | |
---|---|---|
committer | 2017-10-02 13:33:33 -0700 | |
commit | 7b098f62f983738bbf048873b6ecac3b26d40d68 (patch) | |
tree | aa506932bc0babe6c6b1e34c37185143bff5aaa4 /tensorflow/contrib/tensor_forest/python/tensor_forest.py | |
parent | d86104fd3862b26d46fa1d37e0403c6ac32b56ab (diff) |
Clarify expectations about the input_data parameter.
PiperOrigin-RevId: 170745215
Diffstat (limited to 'tensorflow/contrib/tensor_forest/python/tensor_forest.py')
-rw-r--r-- | tensorflow/contrib/tensor_forest/python/tensor_forest.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/tensorflow/contrib/tensor_forest/python/tensor_forest.py b/tensorflow/contrib/tensor_forest/python/tensor_forest.py index 756533250a..eb938763f1 100644 --- a/tensorflow/contrib/tensor_forest/python/tensor_forest.py +++ b/tensorflow/contrib/tensor_forest/python/tensor_forest.py @@ -470,7 +470,11 @@ class RandomForestGraphs(object): """Constructs a TF graph for evaluating a random forest. Args: - input_data: A tensor or dict of string->Tensor for input data. + input_data: A tensor or dict of string->Tensor for the input data. + This input_data must generate the same spec as the + input_data used in training_graph: the dict must have + the same keys, for example, and all tensors must have + the same size in their first dimension. **inference_args: Keyword arguments to pass through to each tree. Returns: |