tf.raw_ops.InTopK

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

tf.raw_ops.InTopK(
    predictions, targets, k, name=None
)

На выходе получается массив batch_size булевых значений, где элемент out[i] равен true, если прогноз для целевого класса входит в число лучших k прогнозов среди всех прогнозов, например i. Обратите внимание, что поведение InTopK отличается от операции TopK в обработке совпадений; если несколько классов имеют одинаковое значение прогноза и находятся на границе лучших k, все эти классы считаются находящимися в числе лучших k.

Более формально, пусть

\(predictions_i\) — прогнозы для всех классов, например i, \(targets_i\) — целевой класс, например i, \(out_i\) — вывод для примера i,