余弦相似度
概述
用向量空间中两个向量夹角的余弦值作为衡量两个个体间差异的大小的度量。余弦值越接近1,就表明夹角越接近0度,也就是两个向量越相似。
一、实现思路
- 分词:对需要比较的文本进行分词,获得词和词频(key/num)。
- 统计:统计所有的词(去重),作为向量的维度。
- 构建:以词作为维度,词频作为当前维度的值,构建向量。没有的以0填充。
- 计算:
二、实现流程
举个简单的例子:
-
准备比较文本
文本一:黑灰化肥灰会挥发发灰黑讳为黑灰花会飞
文本二:灰黑化肥会会挥发发黑灰为讳飞花化为灰
-
分词并统计词频
文本一:“灰”:4,“挥发”:1,“黑”:3,“发”:1,“讳”:1,“化肥”:1,“花会”:1,“会”:1,“为”:1,“飞”:1
文本二:“灰”:3,“挥发”:1,“飞花”:1,“黑”:2,“发”:1,“讳”:1,“化肥”:1,“会”:2,“为”:1,“化为”:1
-
统计所有词
所有词:“灰”,“挥发”,“飞花”,“黑”,“发”,“讳”,“化肥”,“花会”,“会”,“为”,“飞”,“化为”
文本一:(4,1,0,3,1,0,1,1,1,1,0,1,0)
文本二:(3,1,1,2,1,1,1,2,1,0,1,1,1)
-
计算
x = 4 ⋅ 3 + 1 ⋅ 1 + 0 ⋅ 1 + … … + 0 ⋅ 1 + 1 ⋅ 1 + 0 ⋅ 1 4 2 + 1 2 + 0 2 + … … + 0 2 + 1 2 + 0 2 − 3 2 + 1 2 + 1 2 + … … + 1 2 + 1 2 + 1 2 = 0.888336318148377 x = \frac{4\cdot3+1\cdot1+0\cdot1+……+0\cdot1+1\cdot1+0\cdot1}{\sqrt{4^2+1^2+0^2+……+0^2+1^2+0^2}-\sqrt{3^2+1^2+1^2+……+1^2+1^2+1^2}} = 0.888336318148377 x=42+12+02+……+02+12+02−32+12+12+……+12+12+124⋅3+1⋅1+0⋅1+……+0⋅1+1⋅1+0⋅1=0.888336318148377
三、代码实现
代码如下(java):
public class CosAlgorithm {
public static void main(String[] args) {
String str1 = "黑灰化肥灰会挥发发灰黑讳为黑灰花会飞";
String str2 = "灰黑化肥会会挥发发黑灰为讳飞花化为灰";
StopRecognition filter = new StopRecognition();
//过滤掉标点
filter.insertStopNatures("w");
//分词-统计词频
Map<String,Integer> map1= new HashMap<>();
ToAnalysis.parse(str1).recognition(filter).forEach(item -> {
//没有则赋初始值,有则+1
if (map1.get(item.getName()) == null){
map1.put(item.getName(),1);
}else {
map1.put(item.getName(),map1.get(item.getName())+1);
}
});
Map<String,Integer> map2 = new HashMap<>();
ToAnalysis.parse(str2).recognition(filter).forEach(item -> {
//没有则赋初始值,有则+1
if (map2.get(item.getName()) == null){
map2.put(item.getName(),1);
}else {
map2.put(item.getName(),map2.get(item.getName())+1);
}
});
System.out.println("map1="+ JSON.toJSONString(map1));
System.out.println("map2="+ JSON.toJSONString(map2));
Set<String> set1 = map1.keySet();
Set<String> set2 = map2.keySet();
Set<String> setAll = new HashSet<>();
setAll.addAll(set1);
setAll.addAll(set2);
System.out.println("all="+JSON.toJSONString(setAll));
List<Integer> list1 = new ArrayList<>(setAll.size());
List<Integer> list2 = new ArrayList<>(setAll.size());
//构建向量
setAll.forEach(item ->{
if (set1.contains(item)){
list1.add(map1.get(item));
}else {
list1.add(0);
}
if (set2.contains(item)){
list2.add(map2.get(item));
}else {
list2.add(0);
}
});
//计算余弦相似度
int sum =0;
long sq1 = 0;
long sq2 = 0;
double result = 0;
for (int i =0;i<setAll.size();i++){
sum +=list1.get(i)*list2.get(i);
sq1 += list1.get(i)*list1.get(i);
sq2 += list2.get(i)*list2.get(i);
}
result = sum/(Math.sqrt(sq1)*Math.sqrt(sq2));
System.out.println("余弦相似度="+result);
}
}
总结
结合样例,然后运行代码,你就会了~
参考:https://blog.csdn.net/u012160689/article/details/15341303