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\) siano le previsioni per tutte le classi, ad esempio i , \(targets_i\) siano la classe di destinazione, ad esempio i , \(out_i\) siano l'output, ad esempio i ,

$$out_i = predictions_{i, targets_i} TopKIncludingTies(predictions_i)$$

Argomenti:

  • scope: un oggetto Scope
  • previsioni: un tensore classes batch_size x.
  • target: un vettore batch_size di ID di classe.
  • k: numero di elementi principali da considerare per la precisione del calcolo.

Ritorna:

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