public final class BoostedTreesAggregateStats extends PrimitiveOp implements Operand<Float>
The summary stats contains gradients and hessians accumulated for each node, feature dimension id and bucket.
operation| Modifier and Type | Method and Description |
|---|---|
Output<Float> |
asOutput()
Returns the symbolic handle of a tensor.
|
static BoostedTreesAggregateStats |
create(Scope scope,
Operand<Integer> nodeIds,
Operand<Float> gradients,
Operand<Float> hessians,
Operand<Integer> feature,
Long maxSplits,
Long numBuckets)
Factory method to create a class wrapping a new BoostedTreesAggregateStats operation.
|
Output<Float> |
statsSummary()
output Rank 4 Tensor (shape=[splits, feature_dimension, buckets, logits_dimension + hessian_dimension])
containing accumulated stats for each node, feature dimension and bucket.
|
equals, hashCode, op, toStringpublic static BoostedTreesAggregateStats create(Scope scope, Operand<Integer> nodeIds, Operand<Float> gradients, Operand<Float> hessians, Operand<Integer> feature, Long maxSplits, Long numBuckets)
scope - current scopenodeIds - int32; Rank 1 Tensor containing node ids for each example, shape [batch_size].gradients - float32; Rank 2 Tensor (shape=[batch_size, logits_dimension]) with gradients for each example.hessians - float32; Rank 2 Tensor (shape=[batch_size, hessian_dimension]) with hessians for each example.feature - int32; Rank 2 feature Tensors (shape=[batch_size, feature_dimension]).maxSplits - int; the maximum number of splits possible in the whole tree.numBuckets - int; equals to the maximum possible value of bucketized feature.public Output<Float> statsSummary()
public Output<Float> asOutput()
OperandInputs 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 © 2015–2019. All rights reserved.