BoostedTreesTrainingPredict

classe finale pubblica BoostedTreesTrainingPredict

Esegue più predittori di ensemble di regressione additiva su istanze di input e

calcola l'aggiornamento ai logit memorizzati nella cache. È progettato per essere utilizzato durante l'allenamento. Attraversa gli alberi a partire dall'ID dell'albero memorizzato nella cache e dall'ID del nodo memorizzato nella cache e calcola gli aggiornamenti da inviare alla cache.

Metodi pubblici

statico BoostedTreesTrainingPredict
create ( Scope scope, Operando <?> treeEnsembleHandle, Operando <Integer> cachedTreeIds, Operando <Integer> cachedNodeIds, Iterable< Operando <Integer>> bucketizedFeatures, Long logitsDimension)
Metodo factory per creare una classe che racchiude una nuova operazione BoostedTreesTrainingPredict.
Output <Intero>
ID nodo ()
Tensore di rango 1 contenente i nuovi ID dei nodi nei nuovi tree_ids.
Uscita <Mobile>
log parziali ()
Tensore di grado 2 contenente l'aggiornamento dei logit (rispetto ai valori memorizzati nella cache) per ogni esempio.
Output <Intero>
IDalbero ()
Tensore di rango 1 contenente nuovi ID albero per ogni esempio.

Metodi ereditati

Metodi pubblici

public static BoostedTreesTrainingPredict create ( Scope scope, Operando <?> treeEnsembleHandle, Operando <Integer> cachedTreeIds, Operando <Integer> cachedNodeIds, Iterable< Operando <Integer>> bucketizedFeatures, Long logitsDimension)

Metodo factory per creare una classe che racchiude una nuova operazione BoostedTreesTrainingPredict.

Parametri
scopo ambito attuale
cachedTreeIds Tensore di rango 1 contenente gli ID degli alberi memorizzati nella cache che è l'albero di previsione iniziale.
cachedNodeId Tensore di rango 1 contenente l'ID del nodo memorizzato nella cache che è il nodo iniziale della previsione.
Caratteristiche con bucket Un elenco di tensori di rango 1 contenente l'ID bucket per ciascuna funzionalità.
logitsDimension scalare, dimensione dei logit, da utilizzare per la forma dei logit parziali.
ritorna
  • una nuova istanza di BoostedTreesTrainingPredict

output pubblico <intero> nodeId ()

Tensore di rango 1 contenente i nuovi ID dei nodi nei nuovi tree_ids.

Uscita pubblica <Float> partialLogits ()

Tensore di grado 2 contenente l'aggiornamento dei logit (rispetto ai valori memorizzati nella cache) per ogni esempio.

output pubblico <intero> treeIds ()

Tensore di rango 1 contenente nuovi ID albero per ogni esempio.