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,

Аргументы:

• 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