BoostedTreesAggregateStats

classe finale pubblica BoostedTreesAggregateStats

Aggrega il riepilogo delle statistiche accumulate per il batch.

Le statistiche di riepilogo contengono gradienti e iuta accumulati per ciascun nodo, ID dimensione della caratteristica e bucket.

Metodi pubblici

Uscita <Mobile>
comeuscita ()
Restituisce la maniglia simbolica di un tensore.
statico BoostedTreesAggregateStats
create ( scope scope, operando <Integer> nodeId, operando <Float> gradienti, operando <Float> hessian, operando <Integer> funzionalità, maxSplits lunghi, numBuckets lunghi)
Metodo factory per creare una classe che racchiude una nuova operazione BoostedTreesAggregateStats.
Uscita <Mobile>
statisticheRiepilogo ()
output Rango 4 Tensore (shape=[splits, feature_dimension, buckets, logits_dimension + hessian_dimension]) contenente le statistiche accumulate per ogni nodo, dimensione della feature e bucket.

Metodi ereditati

Metodi pubblici

output pubblico <Float> asOutput ()

Restituisce la maniglia simbolica di un tensore.

Gli input per le operazioni TensorFlow sono output di un'altra operazione TensorFlow. Questo metodo viene utilizzato per ottenere un handle simbolico che rappresenta il calcolo dell'input.

creazione statica pubblica BoostedTreesAggregateStats (ambito ambito , operando <Integer> nodeId, operando <Float> gradienti, operando <Float> hessian, operando <Integer> funzionalità, Long maxSplits, Long numBuckets)

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

Parametri
scopo ambito attuale
nodeId int32; Tensore di rango 1 contenente gli ID dei nodi per ciascun esempio, forma [batch_size].
gradienti float32; Tensore di rango 2 (shape=[batch_size, logits_dimension]) con gradienti per ogni esempio.
iuta float32; Tensore di rango 2 (shape=[batch_size, hessian_dimension]) con iuta per ogni esempio.
caratteristica int32; Tensori di feature di rango 2 (shape=[batch_size, feature_dimension]).
maxSplits intero; il numero massimo di suddivisioni possibili nell'intero albero.
numBucket intero; equivale al valore massimo possibile della funzionalità segmentata.
ritorna
  • una nuova istanza di BoostedTreesAggregateStats

output pubblico <Float> statsSummary ()

output Rango 4 Tensore (shape=[splits, feature_dimension, buckets, logits_dimension + hessian_dimension]) contenente le statistiche accumulate per ogni nodo, dimensione della feature e bucket.