tensorflow::ops::InTopK

#include <nn_ops.h>

Определяет, находятся ли целевые значения в числе K лучших прогнозов.

Краткое описание

Это возвращает массив булевых значений batch_size, где запись out[i] равна true, если прогноз для целевого класса находится среди лучших k прогнозов среди всех прогнозов для примера i. Обратите внимание, что поведение InTopK отличается от операции TopK по обработке совпадений; если несколько классов имеют одинаковое значение прогноза и находятся на границе лучших k, все эти классы считаются находящимися в числе лучших k.

Более формально, пусть

\(predictions_i\) — прогнозы для всех классов для примера i, \(targets_i\) — целевой класс для примера i, \(out_i\) — результат для примера i,

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

Аргументы:

• scope: Объект Scope
• predictions: Массив batch_size x classes.
• targets: Вектор batch_size с идентификаторами классов.
• k: Количество лучших элементов для расчета точности.

Возвращает:

• Output: Вычисленная точность в k как bool Tensor.

Публичные атрибуты

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