tensorflow :: opérations :: InTopK

Indique si les cibles sont dans les K principales prédictions.

Résumé

Cela batch_size un tableau batch_size , une entrée out[i] est true si la prédiction pour la classe cible est parmi les k meilleures prédictions parmi toutes les prédictions par exemple i . Notez que le comportement d' InTopK diffère de l'opération TopK dans sa gestion des liens; si plusieurs classes ont la même valeur de prédiction et enjambent le top- k limite, toutes les classes sont considérées comme en haut k .

Plus formellement, laissez

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

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

Arguments:

• scope: un objet Scope
• predictions: Un tenseur de classes batch_size x.
• cibles: un vecteur batch_size d'identifiants de classe.
• k: Nombre d'éléments supérieurs à examiner pour calculer la précision.

Retour:

• Output : précision calculée à k tant que bool Tensor .

Attributs publics

Operation operation

::tensorflow::Output precision

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

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const