tf.keras.losses.KLD
| Просмотр исходного кода на GitHub |
Вычисляет потерю расхождения Кульбака-Лайблера между y_true и y_pred.
tf.keras.losses.KLD(
y_true, y_pred
)
loss = y_true * log(y_true / y_pred)
См.: https://en.wikipedia.org/wiki/Kullback%E2%80%93Leibler_divergence
Использование:
loss = tf.keras.losses.KLD([.4, .9, .2], [.5, .8, .12])
print('Loss: ', loss.numpy()) # Loss: 0.11891246
| Аргументы | |
|---|---|
y_true | Тензор истинных целевых значений. |
y_pred | Тензор предсказанных целевых значений. |
| Возвращаемые значения | |
|---|---|
Объект Tensor с потерей. |
| Исключения | |
|---|---|
TypeError | Если y_true нельзя привести к типу y_pred.dtype. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/keras/losses/KLD