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,

\[out_i = predictions_{i, targets_i} \in TopKIncludingTies(predictions_i)\]