diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-25 09:12:47 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-25 09:17:42 -0700 |
| commit | 32140ae87fd86398ac4fa45cb67bd2f29a93090d (patch) | |
| tree | f7a4d7701a64defd1f0d30a0460a2ce3ea571695 /tensorflow/core/ops | |
| parent | 588787ff7572208285cb471c76f4f8c83ad9d7ec (diff) | |
Boosted trees: Adding categorical split support to prediction ops.
PiperOrigin-RevId: 214448656
Diffstat (limited to 'tensorflow/core/ops')
| -rw-r--r-- | tensorflow/core/ops/boosted_trees_ops.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/core/ops/boosted_trees_ops.cc b/tensorflow/core/ops/boosted_trees_ops.cc index 7c4184bff4..b8cf538554 100644 --- a/tensorflow/core/ops/boosted_trees_ops.cc +++ b/tensorflow/core/ops/boosted_trees_ops.cc @@ -180,6 +180,8 @@ REGISTER_OP("BoostedTreesMakeStatsSummary") return Status::OK(); }); +// TODO(nponomareva): when/if creating the new op for unbucketized data, rename +// bucketized_features to features. REGISTER_OP("BoostedTreesPredict") .Input("tree_ensemble_handle: resource") .Input("bucketized_features: num_bucketized_features * int32") |