tensorflow :: operaciones :: InTopK

#include <nn_ops.h>

Dice si los objetivos se encuentran en las primeras K predicciones.

Resumen

Esto genera una batch_size bool batch_size , una entrada out[i] es true si la predicción para la clase objetivo está entre las k predicciones principales entre todas las predicciones, por ejemplo i . Tenga en cuenta que el comportamiento de InTopK difiere del de la TopK en su manejo de los empates; si varias clases tienen el mismo valor de predicción y se encuentran en el límite superior de k , se considera que todas esas clases están en la parte superior k .

Más formalmente, dejemos

\(predictions_i\) ser las predicciones para todas las clases por ejemplo i , \(targets_i\) ser la clase de destino por ejemplo i , \(out_i\) ser la salida por ejemplo i

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

Argumentos:

  • alcance: un objeto de alcance
  • predicciones: Un tensor de classes batch_size x.
  • batch_size : un vector batch_size de identificadores de clase.
  • k: Número de elementos superiores que se deben tener en cuenta para calcular la precisión.

Devoluciones:

Constructores y Destructores

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

Atributos públicos

operation
precision

Funciones publicas

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

Atributos públicos

operación

Operation operation

precisión

::tensorflow::Output precision

Funciones publicas

InTopK

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

nodo

::tensorflow::Node * node() const

operador :: tensorflow :: Entrada

 operator::tensorflow::Input() const

operador :: tensorflow :: Salida

 operator::tensorflow::Output() const