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.

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