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
