tf.compat.v1.math.in_top_k

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

tf.compat.v1.math.in_top_k(
    predictions, targets, k, name=None
)

Это возвращает массив batch_size типа bool, где элемент out[i] равен true, если прогноз для целевого класса конечен (не inf, -inf или nan) и входит в число 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)\]