org.tensorflow.op.core

Class BoostedTreesMakeStatsSummary

java.lang.Object

org.tensorflow.op.PrimitiveOp

org.tensorflow.op.core.BoostedTreesMakeStatsSummary

All Implemented Interfaces:

Op, Operand<Float>

public final class BoostedTreesMakeStatsSummary
extends PrimitiveOp
implements Operand<Float>

Makes the summary of accumulated stats for the batch.

The summary stats contains gradients and hessians accumulated into the corresponding node and bucket for each example.

Field Summary

Fields inherited from class org.tensorflow.op.PrimitiveOp

operation

Method Summary

Modifier and Type	Method and Description
Output<Float>	asOutput() Returns the symbolic handle of a tensor.
static BoostedTreesMakeStatsSummary	create(Scope scope, Operand<Integer> nodeIds, Operand<Float> gradients, Operand<Float> hessians, Iterable<Operand<Integer>> bucketizedFeaturesList, Long maxSplits, Long numBuckets) Factory method to create a class to wrap a new BoostedTreesMakeStatsSummary operation to the graph.
Output<Float>	statsSummary() output Rank 4 Tensor (shape=[#features, #splits, #buckets, 2]) containing accumulated stats put into the corresponding node and bucket.

Methods inherited from class org.tensorflow.op.PrimitiveOp

equals, hashCode, toString

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Method Detail

create

public static BoostedTreesMakeStatsSummary create(Scope scope,
                                                  Operand<Integer> nodeIds,
                                                  Operand<Float> gradients,
                                                  Operand<Float> hessians,
                                                  Iterable<Operand<Integer>> bucketizedFeaturesList,
                                                  Long maxSplits,
                                                  Long numBuckets)

Factory method to create a class to wrap a new BoostedTreesMakeStatsSummary operation to the graph.

Parameters:

scope - current graph scope

nodeIds - int32 Rank 1 Tensor containing node ids, which each example falls into for the requested layer.

gradients - float32; Rank 2 Tensor (shape=[#examples, 1]) for gradients.

hessians - float32; Rank 2 Tensor (shape=[#examples, 1]) for hessians.

bucketizedFeaturesList - int32 list of Rank 1 Tensors, each containing the bucketized feature (for each feature column).

maxSplits - int; the maximum number of splits possible in the whole tree.

numBuckets - int; equals to the maximum possible value of bucketized feature.

Returns:

a new instance of BoostedTreesMakeStatsSummary

statsSummary

public Output<Float> statsSummary()

output Rank 4 Tensor (shape=[#features, #splits, #buckets, 2]) containing accumulated stats put into the corresponding node and bucket. The first index of 4th dimension refers to gradients, and the second to hessians.

asOutput

public Output<Float> asOutput()

Description copied from interface: Operand

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.

Specified by:

asOutput in interface Operand<Float>

See Also:

OperationBuilder.addInput(Output)