tf.math.in_top_k

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

tf.math.in_top_k(
    targets, predictions, 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)\]