文章/答案/技术大牛

发布

社区首页 >问答首页 >Tensorflow:相当于在tensorflow中学习的pairwise_kernels

问Tensorflow:相当于在tensorflow中学习的pairwise_kernels
EN

Stack Overflow用户

提问于 2022-03-08 13:23:56

回答 1查看 45关注 0票数 0

我试图在两组嵌入之间找到cosine similarity。我使用来自pairwise_kernels的scikit-learn计算它，但现在我试图利用GPU，因此我正在寻找与pairwise_kernels相当的Tensorflow。这就是我使用scikit-learn所做的

sim_elements = pairwise_kernels(input_array, orig_array, metric = 'cosine', n_jobs = -1)

我在网上搜索，偶然发现了tensorflow pairwise_distances，但这不是我想要实现的。我试图实现一对多的相似，而不是一对一的相似。任何帮助都是非常感谢的！

发布于 2022-03-08 13:56:35

如果我理解您的问题正确，我认为您可以使用CosineSimilarity损失函数来自tf.keras.losses：https://www.tensorflow.org/api_docs/python/tf/keras/losses/CosineSimilarity。

但是要小心，他们交换了产出，认为这是一种损失。SO-1是最大相似度和最小相似度.

票数 0

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/71395800

复制

相似问题

问Tensorflow:相当于在tensorflow中学习的pairwise_kernelsEN