Spec-Zone.ru › TensorFlow

tf.keras.losses.cosine_similarity

Вычисляет косинусное сходство между метками и предсказаниями.

tf.keras.losses.cosine_similarity(
    y_true, y_pred, axis=-1
)

Формула:

loss = -sum(l2_norm(y_true) * l2_norm(y_pred))

Обратите внимание, что это число от -1 до 1. Когда это отрицательное число между -1 и 0, 0 указывает на ортогональность, а значения, более близкие к -1, указывают на большее сходство. Это делает его пригодным в качестве функции потерь в случае, когда вы пытаетесь максимизировать близость между предсказаниями и целями. Если либо y_true, либо y_pred является нулевым вектором, косинусное сходство будет равно 0 независимо от близости между предсказаниями и целями.

Аргументы
y_true Tensor истинных целевых значений.
y_pred Tensor предсказанных целевых значений.
axis Ось, по которой определяется сходство. По умолчанию -1.
Возвращаемое значение
Tensor косинусного сходства.

Пример:

y_true = [[0., 1.], [1., 1.], [1., 1.]]
y_pred = [[1., 0.], [1., 1.], [-1., -1.]]
loss = keras.losses.cosine_similarity(y_true, y_pred, axis=-1)
[-0., -0.99999994, 0.99999994]

© 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/losses/cosine_similarity

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API