tf.keras.ops.binary_crossentropy
Вычисляет функцию потерь бинарной кросс-энтропии между целевым тензором и тензором выходных данных.
tf.keras.ops.binary_crossentropy(
target, output, from_logits=False
)
Функция потерь бинарной кросс-энтропии обычно используется в задачах бинарной классификации, где каждый входной образец принадлежит к одному из двух классов. Она измеряет различие между целевыми и выходными вероятностями или логитами.
| Аргументы | |
|---|---|
target | Целевой тензор, представляющий истинные бинарные метки. Его форма должна совпадать с формой тензора output. |
output | Тензор выходных данных, представляющий предсказанные вероятности или логиты. Его форма должна совпадать с формой тензора target. |
from_logits | (необязательно) Указывает, является ли output тензором логитов или вероятностей. Установите значение True, если output представляет логиты; в противном случае установите значение False, если output представляет вероятности. По умолчанию используется значение False. |
| Возвращает | |
|---|---|
Целочисленный тензор: вычисленная функция потерь бинарной кросс-энтропии между target и output. |
Пример:
target = keras.ops.convert_to_tensor([0, 1, 1, 0])
output = keras.ops.convert_to_tensor([0.1, 0.9, 0.8, 0.2])
binary_crossentropy(target, output)
array([0.10536054 0.10536054 0.22314355 0.22314355],
shape=(4,), dtype=float32)
© 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/binary_crossentropy
