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