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,

Аргументы:

• 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