tensorflow cosine相似度实现

最新推荐文章于 2024-08-25 18:49:06 发布

FocusOneThread

最新推荐文章于 2024-08-25 18:49:06 发布

阅读量1w

点赞数

分类专栏： TensorFlow

本文链接：https://blog.csdn.net/guotong1988/article/details/78216650

版权

TensorFlow 专栏收录该内容

230 篇文章 11 订阅 ¥19.90 ¥99.00

订阅专栏

超级会员免费看

本文介绍了在TensorFlow中如何正确实现Cosine相似度，该方法帮助解决了同事代码调试中的问题，且与IRGAN的loss计算方式一致，涉及正例和负例的处理。

摘要由CSDN通过智能技术生成

def cosine(q,a):
    pooled_len_1 = tf.sqrt(tf.reduce_sum(q * q, 1))
    pooled_len_2 = tf.sqrt(tf.reduce_sum(a * a, 1))
    pooled_mul_12 = tf.reduce_sum(q * a, 1)
    score = tf.div(pooled_mul_12, pooled_len_1 * pooled_len_2 +1e-8, name="scores")
    return score

帮一个同事调试代码，最后发现是他的cosine写得有问题，换成这个后好了，loss也换成IRGAN一模一样的了（都是 neg正例和 pos负例都有的），摘自IRGAN
https://github.com/geek-ai/irgan