tensorflow :: operaciones :: InTopKV2
#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 vectorbatch_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 | |
---|---|
InTopKV2 (const :: tensorflow::Scope & scope, :: tensorflow::Input predictions, :: tensorflow::Input targets, :: tensorflow::Input 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
InTopKV2
InTopKV2( const ::tensorflow::Scope & scope, ::tensorflow::Input predictions, ::tensorflow::Input targets, ::tensorflow::Input k )
nodo
::tensorflow::Node * node() const
operador :: tensorflow :: Entrada
operator::tensorflow::Input() const
operador :: tensorflow :: Salida
operator::tensorflow::Output() const