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 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 | |
---|---|
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