tensorflow:: ops:: InTopKV2

Indica si los objetivos se encuentran entre las K predicciones principales.

Resumen

Esto genera una matriz booleana batch_size , una entrada out[i] es true si la predicción para la clase objetivo está entre las k mejores predicciones entre todas las predicciones, por ejemplo i . Tenga en cuenta que el comportamiento de InTopK difiere de la operación TopK en su manejo de empates; si varias clases tienen el mismo valor de predicción y abarcan el límite de k mejores, todas esas clases se consideran entre las k mejores.

Más formalmente, dejemos

\(predictions_i\) sean las predicciones para todas las clases, por ejemplo i , \(targets_i\) sea ​​la clase objetivo, por ejemplo i , \(out_i\) sea ​​la salida por ejemplo i ,

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

Argumentos:

• ámbito: Un objeto de ámbito
• predicciones: Un tensor batch_size x classes .
• objetivos: Un vector batch_size de identificadores de clase.
• k: Número de elementos principales a considerar para calcular la precisión.

Devoluciones:

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

Atributos públicos

Operation operation

::tensorflow::Output precision

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

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const