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 |
---|