тензорный поток :: ops :: InTopK

#include <nn_ops.h>

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

Резюме

Это выводит массив batch_size bool, запись 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
  • предсказания: batch_size classes batch_size x.
  • batch_size вектор идентификаторов классов batch_size.
  • k: количество верхних элементов, на которые следует обратить внимание для определения точности вычислений.

Возврат:

  • Output : вычисленная точность при k как bool Tensor .

Конструкторы и деструкторы

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

Публичные атрибуты

operation
precision

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

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Публичные атрибуты

операция

Operation operation

точность

::tensorflow::Output precision

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

InTopK

 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