BoostedTreesUpdateEnsembleV2

classe finale publique BoostedTreesUpdateEnsembleV2

Met à jour l'ensemble d'arbres en ajoutant une couche au dernier arbre en croissance

ou en démarrant un nouvel arbre.

Classes imbriquées

classe BoostedTreesUpdateEnsembleV2.Options Attributs facultatifs pour BoostedTreesUpdateEnsembleV2

Méthodes publiques

statique BoostedTreesUpdateEnsembleV2
créer ( Scope scope, Operand <?> treeEnsembleHandle, Iterable< Operand <Integer>> featureIds, Iterable< Operand <Integer>> dimensionIds, Iterable< Operand <Integer>> nodeIds, Iterable< Operand <Float>> gains, Iterable< Operand <Integer>> seuils, Iterable< Operand <Float>> leftNodeContribs, Iterable< Operand <Float>> rightNodeContribs, Iterable< Operand <String>> splitTypes, Operand <Integer> maxDepth, Operand <Float> learningRate, Operand <Integer> pruningMode , Options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération BoostedTreesUpdateEnsembleV2.
statique BoostedTreesUpdateEnsembleV2.Options
logitsDimension (LogitsDimension longue)

Méthodes héritées

Méthodes publiques

public static BoostedTreesUpdateEnsembleV2 créer ( Scope scope, Operand <?> treeEnsembleHandle, Iterable< Operand <Integer>> featureIds, Iterable< Operand <Integer>> dimensionIds, Iterable< Operand <Integer>> nodeIds, Iterable< Operand <Float>> gains, Iterable< Operand <Integer>> seuils, Iterable< Operand <Float>> leftNodeContribs, Iterable< Operand <Float>> rightNodeContribs, Iterable< Operand <String>> splitTypes, Operand <Integer> maxDepth, Operand <Float> learningRate, Operand < Entier> Mode d'élagage, Options... options)

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

Paramètres
portée portée actuelle
treeEnsembleHandle Handle de la variable d’ensemble.
identifiants de fonctionnalités Tenseur de rang 1 avec identifiants pour chaque fonctionnalité. Il s'agit du véritable identifiant de la fonctionnalité qui sera utilisée dans le fractionnement.
identifiants de dimension Liste des tenseurs de rang 1 représentant la dimension dans chaque fonctionnalité.
ID de nœud Liste des tenseurs de rang 1 représentant les nœuds pour lesquels cette fonctionnalité présente un split.
gains Liste des tenseurs de rang 1 représentant les gains pour chacune des divisions de la fonctionnalité.
seuils Liste des tenseurs de rang 1 représentant les seuils pour chacune des divisions de la fonctionnalité.
leftNodeContribs Liste des tenseurs de rang 2 avec les contributions de la feuille de gauche pour chacune des divisions de la fonctionnalité. Sera ajouté aux valeurs des nœuds précédents pour constituer les valeurs des nœuds de gauche.
rightNodeContribs Liste des tenseurs de rang 2 avec les contributions de la feuille droite pour chacune des divisions de la fonctionnalité. Sera ajouté aux valeurs des nœuds précédents pour constituer les valeurs des bons nœuds.
Types divisés Liste des tenseurs de rang 1 représentant le type de division pour chaque fonctionnalité.
profondeur max Profondeur maximale de l'arbre à construire.
taux d'apprentissage const de retrait pour chaque nouvel arbre.
Mode d'élagage 0-Pas de taille, 1-Pré-taille, 2-Post-taille.
choix porte des valeurs d'attributs facultatifs
Retour
  • une nouvelle instance de BoostedTreesUpdateEnsembleV2

public statique BoostedTreesUpdateEnsembleV2.Options logitsDimension (Long logitsDimension)

Paramètres
logitsDimension scalaire, dimension des logits