tf.raw_ops.BoostedTreesAggregateStats | TensorFlow v2.16.1 (original) (raw)
tf.raw_ops.BoostedTreesAggregateStats
Stay organized with collections Save and categorize content based on your preferences.
Aggregates the summary of accumulated stats for the batch.
View aliases
Compat aliases for migration
SeeMigration guide for more details.
tf.compat.v1.raw_ops.BoostedTreesAggregateStats
tf.raw_ops.BoostedTreesAggregateStats(
node_ids, gradients, hessians, feature, max_splits, num_buckets, name=None
)
The summary stats contains gradients and hessians accumulated for each node, feature dimension id and bucket.
| 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 | A Tensor of type int32. int32; Rank 2 feature Tensors (shape=[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. |
| name | A name for the operation (optional). |
| Returns |
|---|
| A Tensor of type float32. |
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.
Last updated 2024-04-26 UTC.