当前位置:   article > 正文

衡量两个向量相似度的方法:余弦相似度、pytorch 求余弦相似度:torch.nn.CosineSimilarity(dim=1, eps=1e-08)

nn.cosinesimilarity

余 弦 相 似 度 \color{red}{余弦相似度}
NLP的任务里,会对生成两个词向量进行相似度的计算,常常采用余弦相似度公式计算。

余弦相似度用向量空间中两个向量夹角的余弦值作为衡量两个个体间差异的大小。余弦值越接近1,就表明夹角越接近0度,也就是两个向量越相似,这就叫"余弦相似性"。
在这里插入图片描述
我们知道,对于两个向量,如果他们之间的夹角越小,那么我们认为这两个向量是越相似的。余弦相似性就是利用了这个理论思想。它通过计算两个向量的夹角的余弦值来衡量向量之间的相似度值。

公式如下:
在这里插入图片描述
参考:衡量两个向量相似度的方法:余弦相似度

用于torch的余弦相似度函数:
pytorch 求余弦相似度:torch.nn.CosineSimilarity(dim=1, eps=1e-08)

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/article/detail/42238
推荐阅读
相关标签
  

闽ICP备14008679号