W3cubDocs

/TensorFlow 2.4

tf.raw_ops.BoostedTreesPredict

Runs multiple additive regression ensemble predictors on input instances and

computes the logits. It is designed to be used during prediction. It traverses all the trees and calculates the final score for each instance.

Args
tree_ensemble_handle A Tensor of type resource.
bucketized_features A list of at least 1 Tensor objects with type int32. A list of rank 1 Tensors containing bucket id for each feature.
logits_dimension An int. scalar, dimension of the logits, to be used for partial logits shape.
name A name for the operation (optional).
Returns
A Tensor of type float32.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/raw_ops/BoostedTreesPredict