tf.raw_ops.BoostedTreesMakeStatsSummary
Makes the summary of accumulated stats for the batch.
tf.raw_ops.BoostedTreesMakeStatsSummary( node_ids, gradients, hessians, bucketized_features_list, max_splits, num_buckets, name=None )
The summary stats contains gradients and hessians accumulated into the corresponding node and bucket for each example.
Args | |
---|---|
node_ids | A Tensor of type int32 . int32 Rank 1 Tensor containing node ids, which each example falls into for the requested layer. |
gradients | A Tensor of type float32 . float32; Rank 2 Tensor (shape=[#examples, 1]) for gradients. |
hessians | A Tensor of type float32 . float32; Rank 2 Tensor (shape=[#examples, 1]) for hessians. |
bucketized_features_list | A list of at least 1 Tensor objects with type int32 . int32 list of Rank 1 Tensors, each containing the bucketized feature (for each feature column). |
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. |
name | A name for the operation (optional). |
Returns | |
---|---|
A Tensor of type float32 . |
© 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/BoostedTreesMakeStatsSummary