ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

TensorForestTreePredict

public final class TensorForestTreePredict

Output the logits for the given input data

Public Methods

Output<Float>
asOutput()
Returns the symbolic handle of a tensor.
static TensorForestTreePredict
create(Scope scope, Operand<?> treeHandle, Operand<Float> denseFeatures, Long logitsDimension)
Factory method to create a class wrapping a new TensorForestTreePredict operation.
Output<Float>
logits()
The logits predictions from the tree for each instance in the batch.

Inherited Methods

Public Methods

public Output<Float> asOutput ()

Returns the symbolic handle of a tensor.

Inputs to TensorFlow operations are outputs of another TensorFlow operation. This method is used to obtain a symbolic handle that represents the computation of the input.

public static TensorForestTreePredict create (Scope scope, Operand<?> treeHandle, Operand<Float> denseFeatures, Long logitsDimension)

Factory method to create a class wrapping a new TensorForestTreePredict operation.

Parameters
scope current scope
treeHandle Handle to the tree resource.
denseFeatures Rank 2 dense features tensor.
logitsDimension Scalar, dimension of the logits.
Returns
  • a new instance of TensorForestTreePredict

public Output<Float> logits ()

The logits predictions from the tree for each instance in the batch.