BoostedTreesCalculateBestGainsPerFeature

classe finale publique BoostedTreesCalculateBestGainsPerFeature

Calcule les gains pour chaque fonctionnalité et renvoie les meilleures informations de répartition possibles pour la fonctionnalité.

Les informations de répartition correspondent au meilleur seuil (ID de compartiment), aux gains et aux contributions des nœuds gauche/droite par nœud pour chaque fonctionnalité.

Il est possible que tous les nœuds ne puissent pas être divisés sur chaque entité. Par conséquent, la liste des nœuds possibles peut différer selon les fonctionnalités. Par conséquent, nous renvoyons `node_ids_list` pour chaque fonctionnalité, contenant la liste des nœuds que cette fonctionnalité peut être utilisée pour diviser.

De cette manière, le résultat est la meilleure répartition par fonctionnalités et par nœud, de sorte qu'il doit être combiné ultérieurement pour produire la meilleure répartition pour chaque nœud (parmi toutes les fonctionnalités possibles).

La longueur des listes de sortie est toutes de la même longueur, `num_features`. Les formes de sortie sont compatibles de manière à ce que la première dimension de tous les tenseurs de toutes les listes soit la même et égale au nombre de nœuds divisés possibles pour chaque fonctionnalité.

Constantes

Chaîne OP_NAME Le nom de cette opération, tel que connu par le moteur principal TensorFlow

Méthodes publiques

statique BoostedTreesCalculateBestGainsPerFeature
créer ( Scope scope, Operand < TInt32 > nodeIdRange, Iterable < Operand < TFloat32 >> statsSummaryList, Operand < TFloat32 > l1, Operand < TFloat32 > l2, Operand < TFloat32 > treeComplexity, Operand < TFloat32 > minNodeWeight, Long maxSplits)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération BoostedTreesCalculateBestGainsPerFeature.
Liste< Sortie < TFloat32 >>
gainsListe ()
Une liste de sortie de tenseurs de rang 1 indiquant les meilleurs gains pour chaque fonctionnalité à diviser pour certains nœuds.
Liste< Sortie < TFloat32 >>
leftNodeContribsList ()
Une liste de tenseurs de rang 2 indiquant la contribution des nœuds de gauche lors du branchement des nœuds parents (donnés par l'élément tenseur dans la sortie node_ids_list) vers la direction gauche par le seuil donné pour chaque fonctionnalité.
Liste< Sortie < TInt32 >>
nodeIdsList ()
Une liste de sortie de tenseurs de rang 1 indiquant les identifiants de nœuds divisés possibles pour chaque fonctionnalité.
Liste< Sortie < TFloat32 >>
rightNodeContribsList ()
Une liste de tenseurs de rang 2, avec les mêmes forme/conditions que left_node_contribs_list, mais juste que la valeur soit pour le nœud droit.
Liste< Sortie < TInt32 >>
seuilsListe ()
Une liste de sortie de tenseurs de rang 1 indiquant l'identifiant du bucket avec lequel comparer (comme seuil) pour la division dans chaque nœud.

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

Valeur constante : "BoostedTreesCalculateBestGainsPerFeature"

Méthodes publiques

public static BoostedTreesCalculateBestGainsPerFeature créer ( Scope scope, Operand < TInt32 > nodeIdRange, Iterable < Operand < TFloat32 >> statsSummaryList, Operand < TFloat32 > l1, Operand < TFloat32 > l2, Operand < TFloat32 > treeComplexity, Operand < TFloat32 > minNodeWeight, Long maxSplits)

Méthode d'usine pour créer une classe encapsulant une nouvelle opération BoostedTreesCalculateBestGainsPerFeature.

Paramètres
portée portée actuelle
nodeIdRange Un tenseur de rang 1 (shape=[2]) pour spécifier la plage [premier, dernier) des identifiants de nœuds à traiter dans `stats_summary_list`. Les nœuds sont itérés entre les deux nœuds spécifiés par le tenseur, comme « pour node_id in range(node_id_range[0], node_id_range[1]) » (notez que le dernier index node_id_range[1] est exclusif).
statsRésuméListe Une liste de tenseurs de rang 3 (#shape=[max_splits, bucket, 2]) pour le résumé des statistiques accumulées (gradient/hessian) par nœud et par bucket pour chaque fonctionnalité. La première dimension du tenseur est le nombre maximum de divisions, et donc tous les éléments ne seront pas utilisés, mais seuls les index spécifiés par node_ids seront utilisés.
l1 l1 facteur de régularisation sur le poids des feuilles, par instance.
l2 Facteur de régularisation l2 sur le poids des feuilles, basé sur chaque instance.
arbreComplexité ajustement du gain, par vantail.
minNodeWeight moyenne minimale de hessiens dans un nœud avant d'être requise pour que le nœud soit pris en compte pour le fractionnement.
maxSplits le nombre de nœuds qui peuvent être divisés dans l'ensemble de l'arborescence. Utilisé comme dimension des tenseurs de sortie.
Retour
  • une nouvelle instance de BoostedTreesCalculateBestGainsPerFeature

liste publique < Sortie < TFloat32 >> gainsList ()

Une liste de sortie de tenseurs de rang 1 indiquant les meilleurs gains pour chaque fonctionnalité à diviser pour certains nœuds. Voir ci-dessus pour plus de détails comme les formes et les tailles.

liste publique < Sortie < TFloat32 >> leftNodeContribsList ()

Une liste de tenseurs de rang 2 indiquant la contribution des nœuds de gauche lors du branchement des nœuds parents (donnés par l'élément tenseur dans la sortie node_ids_list) vers la direction gauche par le seuil donné pour chaque fonctionnalité. Cette valeur sera utilisée pour créer la valeur du nœud gauche en l'ajoutant à la valeur du nœud parent. La taille de la deuxième dimension est de 1 pour les logits unidimensionnels, mais serait plus grande pour les problèmes multi-classes. Voir ci-dessus pour plus de détails comme les formes et les tailles.

liste publique < Sortie < TInt32 >> nodeIdsList ()

Une liste de sortie de tenseurs de rang 1 indiquant les identifiants de nœuds divisés possibles pour chaque fonctionnalité. La longueur de la liste est num_features, mais chaque tenseur a une taille différente car chaque fonctionnalité fournit différents nœuds possibles. Voir ci-dessus pour plus de détails comme les formes et les tailles.

liste publique < Sortie < TFloat32 >> rightNodeContribsList ()

Une liste de tenseurs de rang 2, avec les mêmes forme/conditions que left_node_contribs_list, mais juste que la valeur soit pour le nœud droit.

liste publique < Sortie < TInt32 >> seuilsList ()

Une liste de sortie de tenseurs de rang 1 indiquant l'identifiant du bucket avec lequel comparer (comme seuil) pour la division dans chaque nœud. Voir ci-dessus pour plus de détails comme les formes et les tailles.