tf.raw_ops.BoostedTreesSparseAggregateStats

Aggregates the summary of accumulated stats for the batch.

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

Args
node_ids A Tensor of type int32. int32; Rank 1 Tensor containing node ids for each example, shape [batch_size].
gradients A Tensor of type float32. float32; Rank 2 Tensor (shape=[batch_size, logits_dimension]) with gradients for each example.
hessians A Tensor of type float32. float32; Rank 2 Tensor (shape=[batch_size, hessian_dimension]) with hessians for each example.
feature_indices A 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_values A 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_shape A 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_splits An int that is >= 1. int; the maximum number of splits possible in the whole tree.
num_buckets An int that is >= 1. int; equals to the maximum possible value of bucketized feature + 1.
name A name for the operation (optional).
Returns
A tuple of Tensor objects (stats_summary_indices, stats_summary_values, stats_summary_shape).
stats_summary_indices A Tensor of type int32.
stats_summary_values A Tensor of type float32.
stats_summary_shape A Tensor of type int32.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/raw_ops/BoostedTreesSparseAggregateStats