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,

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