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.