BoostedTreesUpdateEnsemble

classe finale publique BoostedTreesUpdateEnsemble

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

ou en démarrant un nouvel arbre.

Méthodes publiques

statique BoostedTreesUpdateEnsemble
créer ( Scope scope, Operand <?> treeEnsembleHandle, Operand <Integer> featureIds, Iterable< Operand <Integer>> nodeIds, Iterable< Operand <Float>> gains, Iterable< Operand <Integer>> seuils, Iterable< Operand <Float> > leftNodeContribs, Iterable< Operand <Float>> rightNodeContribs, Operand <Integer> maxDepth, Operand <Float> learningRate, Long pruningMode)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération BoostedTreesUpdateEnsemble.

Méthodes héritées

Méthodes publiques

public static BoostedTreesUpdateEnsemble créer ( Scope scope, Operand <?> treeEnsembleHandle, Operand <Integer> featureIds, Iterable< Operand <Integer>> nodeIds, Iterable< Operand <Float>> gains, Iterable< Operand <Integer>> seuils, Iterable< Operand <Float>> leftNodeContribs, Iterable< Operand <Float>> rightNodeContribs, Operand <Integer> maxDepth, Operand <Float> learningRate, Long pruningMode)

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

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.
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.
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.
Retour
  • une nouvelle instance de BoostedTreesUpdateEnsemble