public final class BoostedTreesPredict extends PrimitiveOp implements Operand<Float>
computes the logits. It is designed to be used during prediction. It traverses all the trees and calculates the final score for each instance.
operation
Modifier and Type | Method and Description |
---|---|
Output<Float> |
asOutput()
Returns the symbolic handle of a tensor.
|
static BoostedTreesPredict |
create(Scope scope,
Operand<?> treeEnsembleHandle,
Iterable<Operand<Integer>> bucketizedFeatures,
Long logitsDimension)
Factory method to create a class wrapping a new BoostedTreesPredict operation.
|
Output<Float> |
logits()
Output rank 2 Tensor containing logits for each example.
|
equals, hashCode, op, toString
public static BoostedTreesPredict create(Scope scope, Operand<?> treeEnsembleHandle, Iterable<Operand<Integer>> bucketizedFeatures, Long logitsDimension)
scope
- current scopetreeEnsembleHandle
- bucketizedFeatures
- A list of rank 1 Tensors containing bucket id for each
feature.logitsDimension
- scalar, dimension of the logits, to be used for partial logits
shape.public Output<Float> asOutput()
Operand
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.
asOutput
in interface Operand<Float>
OperationBuilder.addInput(Output)
Copyright © 2022. All rights reserved.