tf.keras.ops.in_top_k
Проверяет, входят ли целевые значения в число k лучших предсказаний.
tf.keras.ops.in_top_k(
targets, predictions, k
)
| Аргументы | |
|---|---|
targets | Тензор истинных меток. |
predictions | Тензор предсказанных меток. |
k | Целое число, задающее количество учитываемых предсказаний. |
| Возвращает | |
|---|---|
Булев тензор той же формы, что и targets, в котором каждый элемент указывает, входит ли соответствующее целевое значение в число k лучших предсказаний. |
Пример:
targets = keras.ops.convert_to_tensor([2, 5, 3]) predictions = keras.ops.convert_to_tensor( [[0.1, 0.4, 0.6, 0.9, 0.5], [0.1, 0.7, 0.9, 0.8, 0.3], [0.1, 0.6, 0.9, 0.9, 0.5]]) in_top_k(targets, predictions, k=3) array([ True False True], shape=(3,), dtype=bool)
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/keras/ops/in_top_k
