tf.keras.backend.in_top_k(
predictions, targets, k
)
Аргументы
predictions
Массив формы (batch_size, classes) и типа float32.
targets
Одномерный массив длины batch_size и типа int32 или int64.
k
Целое число, количество лучших элементов для рассмотрения.
Возвращаемое значение
Одномерный массив длины batch_size и типа bool. output[i] имеет значение True, если predictions[i, targets[i]] находится среди k лучших значений predictions[i].