flux tensoriel : : opérations : : DansTopK

Indique si les cibles figurent dans les K premières prédictions.

Résumé

Cela génère un tableau booléen batch_size , une entrée out[i] est true si la prédiction pour la classe cible fait partie des 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 chevauchent la limite top- k , toutes ces classes sont considérées comme faisant partie du top k .

Plus formellement, disons

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

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

Arguments :

• scope : 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 à prendre en compte pour la précision du calcul.

Retours :

• Output : Précision calculée à k sous forme de 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