tensorflow::ops::InTopK

#include <nn_ops.h>

Indique si les cibles figurent parmi les K meilleures prédictions.

Résumé

Cette fonction renvoie un tableau booléen batch_size . L'entrée out[i] est true si la prédiction pour la classe cible figure parmi les k meilleures prédictions parmi toutes les prédictions, par exemple i . Notez que le comportement de InTopK diffère de celui de l'opération TopK dans sa gestion des égalités ; si plusieurs classes ont la même valeur de prédiction et chevauchent la limite k meilleures, toutes ces classes sont considérées comme faisant partie des k meilleures.

Plus formellement, laissez

\(predictions_i\) soient les prédictions pour toutes les classes par exemple i , \(targets_i\) soit la classe cible par exemple i , \(out_i\) soit la sortie par exemple i ,

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

Arguments :

  • portée : un objet Scope
  • prédictions : un tenseur batch_size x classes .
  • cibles : un vecteur batch_size d’identifiants de classe.
  • k : Nombre d'éléments principaux à examiner pour calculer la précision.

Retours :

  • Output : Précision calculée à k sous forme de bool Tensor .

Constructeurs et destructeurs

InTopK (const :: tensorflow::Scope & scope, :: tensorflow::Input predictions, :: tensorflow::Input targets, int64 k)

attributs publics

operation
precision

fonctions publiques

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

attributs publics

opération 

Operation operation

précision 

::tensorflow::Output precision

fonctions publiques

InTopK 

 InTopK(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input predictions,
  ::tensorflow::Input targets,
  int64 k
)

nœud 

::tensorflow::Node * node() const

opérateur::tensorflow::Entrée 

 operator::tensorflow::Input() const

opérateur::tensorflow::Sortie 

 operator::tensorflow::Output() const