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).
[[["Easy to understand","easyToUnderstand","thumb-up"],["Solved my problem","solvedMyProblem","thumb-up"],["Other","otherUp","thumb-up"]],[["Missing the information I need","missingTheInformationINeed","thumb-down"],["Too complicated / too many steps","tooComplicatedTooManySteps","thumb-down"],["Out of date","outOfDate","thumb-down"],["Samples / code issue","samplesCodeIssue","thumb-down"],["Other","otherDown","thumb-down"]],["Last updated 2024-01-23 UTC."],[],[]]