tf.compat.v1.math.in_top_k

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

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

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

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

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