diff options
Diffstat (limited to 'tensorflow/contrib/tensor_forest/hybrid')
| -rw-r--r-- | tensorflow/contrib/tensor_forest/hybrid/core/ops/stochastic_hard_routing_function_op.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/tensor_forest/hybrid/core/ops/stochastic_hard_routing_function_op.cc b/tensorflow/contrib/tensor_forest/hybrid/core/ops/stochastic_hard_routing_function_op.cc index 09b83e2af1..66aa293dc1 100644 --- a/tensorflow/contrib/tensor_forest/hybrid/core/ops/stochastic_hard_routing_function_op.cc +++ b/tensorflow/contrib/tensor_forest/hybrid/core/ops/stochastic_hard_routing_function_op.cc @@ -70,7 +70,7 @@ REGISTER_OP("StochasticHardRoutingFunction") return Status::OK(); }) .Doc(R"doc( - Samples a path for each instance in `input_data` and returns the + Samples a path for each instance in `input_data` and returns the probability of the path and the path taken. tree_depth: The depth of the decision tree. |