BoostedTreesUpdateEnsemble

public final class BoostedTreesUpdateEnsemble

Aggiorna l'insieme di alberi aggiungendo un livello all'ultimo albero in crescita

o iniziando un nuovo albero.

Metodi pubblici

static BoostedTreesUpdateEnsemble
create ( Scope scope, Operand <?> treeEnsembleHandle, Operand <Integer> featureIds, Iterable < Operand <Integer>> nodeIds, Iterable < Operand <Float>> gains, Iterable < Operand <Integer>> thresholds, Iterable < Operand <Float> > leftNodeContribs, Iterable < Operand <Float>> rightNodeContribs, Operand <Integer> maxDepth, Operand <Float> learningRate, Long pruningMode)
Metodo Factory per creare una classe che racchiude una nuova operazione BoostedTreesUpdateEnsemble.

Metodi ereditati

Metodi pubblici

public static BoostedTreesUpdateEnsemble create ( Scope scope, Operand <?> treeEnsembleHandle, Operand <Integer> featureIds, Iterable < Operand <Integer>> nodeIds, Iterable < Operand <Float>> gains, Iterable < Operand <Integer>> soglie, Iterable < Operand <Float>> leftNodeContribs, Iterable < Operando <Float>> rightNodeContribs, Operando <Integer> maxDepth, Operando <Float> learningRate, Long pruningMode)

Metodo Factory per creare una classe che racchiude una nuova operazione BoostedTreesUpdateEnsemble.

Parametri
scopo ambito attuale
treeEnsembleHandle Gestire la variabile dell'insieme.
featureIds Rango 1 tensore con ID per ogni caratteristica. Questo è l'ID reale della funzione che verrà utilizzata nella divisione.
nodeIds Elenco dei tensori di rango 1 che rappresentano i nodi per i quali questa caratteristica ha una divisione.
guadagni Elenco dei tensori di rango 1 che rappresentano i guadagni per ciascuna divisione della funzione.
soglie Elenco dei tensori di rango 1 che rappresentano le quote per ciascuna divisione della caratteristica.
leftNodeContribs Elenco dei tensori di rango 2 con contributi foglia sinistra per ciascuna delle suddivisioni della caratteristica. Verrà aggiunto ai valori del nodo precedente per costituire i valori dei nodi di sinistra.
rightNodeContribs Elenco dei tensori di rango 2 con contributi foglia destra per ciascuna delle suddivisioni della caratteristica. Verrà aggiunto ai valori del nodo precedente per costituire i valori dei nodi giusti.
profondità massima Profondità massima dell'albero da costruire.
learningRate const di ritiro per ogni nuovo albero.
potaturaMode 0-Nessuna potatura, 1-Pre-potatura, 2-Post-potatura.
ritorna
  • una nuova istanza di BoostedTreesUpdateEnsemble