tensoreflusso:: ops:: InTopK
#include <nn_ops.h> Indica se gli obiettivi rientrano nelle previsioni K principali.
Riepilogo
Ciò restituisce un array bool batch_size , una voce out[i] è true se la previsione per la classe di destinazione è tra le prime k previsioni tra tutte le previsioni, ad esempio i . Si noti che il comportamento di InTopK differisce dall'operazione TopK nella gestione dei valori di parità; se più classi hanno lo stesso valore di previsione e si trovano a cavallo del limite top- k , tutte quelle classi sono considerate nel top k .
Più formalmente, lasciamo
\(predictions_i\) essere le previsioni per tutte le classi, ad esempio i , \(targets_i\) essere la classe di destinazione, ad esempio i , \(out_i\) essere l'output per esempio i ,
Argomenti:
- scope: un oggetto Scope
- previsioni: un tensore
classesbatch_sizex. - target: un vettore
batch_sizedi ID di classe. - k: numero di elementi principali da considerare per la precisione del calcolo.
Resi:
Costruttori e distruttori | |
|---|---|
InTopK (const :: tensorflow::Scope & scope, :: tensorflow::Input predictions, :: tensorflow::Input targets, int64 k) |
Attributi pubblici | |
|---|---|
operation | |
precision | |
Funzioni pubbliche | |
|---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const | |
Attributi pubblici
operazione
Operation operation
precisione
::tensorflow::Output precision
Funzioni pubbliche
InTopK
InTopK( const ::tensorflow::Scope & scope, ::tensorflow::Input predictions, ::tensorflow::Input targets, int64 k )
nodo
::tensorflow::Node * node() const
operatore::tensorflow::Input
operator::tensorflow::Input() const
operatore::tensorflow::Output
operator::tensorflow::Output() const