tf.raw_ops.BoostedTreesSparseAggregateStats

Aggregates the summary of accumulated stats for the batch.

tf.raw_ops.BoostedTreesSparseAggregateStats(
    node_ids,
    gradients,
    hessians,
    feature_indices,
    feature_values,
    feature_shape,
    max_splits,
    num_buckets,
    name=None
)

The summary stats contains gradients and hessians accumulated for each node, bucket and dimension id.

node_idsA Tensor of type int32. int32; Rank 1 Tensor containing node ids for each example, shape [batch_size].
gradientsA Tensor of type float32. float32; Rank 2 Tensor (shape=[batch_size, logits_dimension]) with gradients for each example.
hessiansA Tensor of type float32. float32; Rank 2 Tensor (shape=[batch_size, hessian_dimension]) with hessians for each example.
feature_indicesA Tensor of type int32. int32; Rank 2 indices of feature sparse Tensors (shape=[number of sparse entries, 2]). Number of sparse entries across all instances from the batch. The first value is the index of the instance, the second is dimension of the feature. The second axis can only have 2 values, i.e., the input dense version of Tensor can only be matrix.
feature_valuesA Tensor of type int32. int32; Rank 1 values of feature sparse Tensors (shape=[number of sparse entries]). Number of sparse entries across all instances from the batch. The first value is the index of the instance, the second is dimension of the feature.
feature_shapeA Tensor of type int32. int32; Rank 1 dense shape of feature sparse Tensors (shape=[2]). The first axis can only have 2 values, [batch_size, feature_dimension].
max_splitsAn int that is >= 1. int; the maximum number of splits possible in the whole tree.
num_bucketsAn int that is >= 1. int; equals to the maximum possible value of bucketized feature + 1.
nameA name for the operation (optional).

A tuple of Tensor objects (stats_summary_indices, stats_summary_values, stats_summary_shape).
stats_summary_indicesA Tensor of type int32.
stats_summary_valuesA Tensor of type float32.
stats_summary_shapeA Tensor of type int32.