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