gpt4 book ai didi

python - Keras CosineSimilarity - 正或负

转载 作者:行者123 更新时间:2023-12-04 09:42:06 24 4
gpt4 key购买 nike

我正在训练一个模型,我的损失函数是余弦相似度:

model.compile(optimizer='adam', 
loss=tf.keras.losses.cosine_similarity,
metrics=[tf.keras.metrics.CosineSimilarity(axis=1)])

在 Keras 中,“余弦相似度” loss应该收敛到 -1 ?

另一方面,“余弦相似度” metric应该去 1 为了更好的准确性,对吧?

提前致谢!

最佳答案

因为这是作为 实现的亏损我们最小化 亏损在训练期间,符号是倒置的,因此当您最小化 时亏损 (朝向-1),实际相似度被最大化到 1。我也认为稳定版本中的文档有点困惑,参见 Nightly documentation用这个例子:

y_true = [[0., 1.], [1., 1.], [1., 1.]]
y_pred = [[1., 0.], [1., 1.], [-1., -1.]]
loss = tf.keras.losses.cosine_similarity(y_true, y_pred, axis=1)
loss.numpy()
array([-0., -0.999, 0.999], dtype=float32)

出于某种原因, 公制 也用 -(减号)实现。

关于python - Keras CosineSimilarity - 正或负,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/62280366/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com