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