tensorflow::ops::InTopK

#include <nn_ops.h>

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

Summary

Этот метод возвращает массив значений типа 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} TopKIncludingTies(predictions_i)$$

Аргументы:

• scope: Объект Scope
• predictions: Тензор batch_size x classes.
• targets: Вектор с идентификаторами классов.
• 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