tensorflow :: operaciones :: InTopK

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

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:

• Output : Precisión calculada en k como un bool Tensor .

Atributos públicos

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