Agrége le résumé des statistiques accumulées pour le lot.
Les statistiques récapitulatives contiennent les dégradés et les hessiens accumulés pour chaque nœud, l'ID de dimension de fonctionnalité et le compartiment.
Constantes
Chaîne | OP_NAME | Le nom de cette opération, tel que connu par le moteur principal TensorFlow |
Méthodes publiques
Sortie < TFloat32 > | comme Sortie () Renvoie le handle symbolique du tenseur. |
statique BoostedTreesAggregateStats | |
Sortie < TFloat32 > | statsRésumé () Tensor de rang 4 de sortie (shape=[splits, feature_dimension, buckets, logits_dimension + hessian_dimension]) contenant les statistiques accumulées pour chaque nœud, dimension de fonctionnalité et bucket. |
Méthodes héritées
Constantes
chaîne finale statique publique OP_NAME
Le nom de cette opération, tel que connu par le moteur principal TensorFlow
Méthodes publiques
Sortie publique < TFloat32 > asOutput ()
Renvoie le handle symbolique du tenseur.
Les entrées des opérations TensorFlow sont les sorties d'une autre opération TensorFlow. Cette méthode est utilisée pour obtenir un handle symbolique qui représente le calcul de l’entrée.
public static BoostedTreesAggregateStats créer ( Portée d'application , Opérande < TInt32 > nodeIds, Opérande < TFloat32 > gradients, Opérande < TFloat32 > hessiens, Opérande < TInt32 > fonctionnalité, Long maxSplits, Long numBuckets)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération BoostedTreesAggregateStats.
Paramètres
portée | portée actuelle |
---|---|
ID de nœud | int32; Tenseur de rang 1 contenant les identifiants de nœuds pour chaque exemple, forme [batch_size]. |
dégradés | flotteur32 ; Tenseur de rang 2 (shape=[batch_size, logits_dimension]) avec des dégradés pour chaque exemple. |
hessiens | flotteur32 ; Tenseur de rang 2 (shape=[batch_size, hessian_dimension]) avec des toiles de jute pour chaque exemple. |
fonctionnalité | int32; Tenseurs de fonctionnalités de rang 2 (shape=[batch_size, feature_dimension]). |
maxSplits | entier ; le nombre maximum de divisions possibles dans l'ensemble de l'arborescence. |
nombre de seaux | entier ; est égal à la valeur maximale possible de la fonctionnalité compartimentée. |
Retour
- une nouvelle instance de BoostedTreesAggregateStats