tf.raw_ops.BoostedTreesPredict

Runs multiple additive regression ensemble predictors on input instances and

tf.raw_ops.BoostedTreesPredict(
    tree_ensemble_handle, bucketized_features, logits_dimension, name=None
)

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

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).

A Tensor of type float32.