tensorflow::ops::InTopKV2

#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

Публичные функции

 InTopKV2(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input predictions,
  ::tensorflow::Input targets,
  ::tensorflow::Input k
)

::tensorflow::Node * node() const 

operator::tensorflow::Input() const 

operator::tensorflow::Output() const