Java代码实现余弦相似度算法比较两字符串相似度

因工作需要比较两个两个字符串的相似度比较,由于最短编辑距离算法不符合需求,就又找其他算法,在网上看到了另一个算法:余弦相似度算法。于是学习了一下,并写篇博客保存,以便学习以及日后用到。
代码如下:

import java.util.HashMap
import java.util.Map;
import java.util.Set;
  


  /*
   * 计算两个字符串(英文字符)的相似度,简单的余弦计算,未添权重
   */
   

public class AnotherCompare {
Map vectorMap = new HashMap();   

int[] tempArray = null;

public AnotherCompare(String string1, String string2) {

    for (Character character1 : string1.toCharArray()) {  //拆解为向量
        if (vectorMap.containsKey(character1)) {   //该字符是否已经存在map中
            vectorMap.get(character1)[0]++;   //key对应的值是一个数组,第一个字符串的字符出现次数存储在数组第一个位置
        } else {
            tempArray = new int[2];  
            tempArray[0] = 1;  //如果字符没有出现过,就添加到map中,数组第一个位置设为1
            tempArray[1] = 0;
            vectorMap.put(character1, tempArray);
        }
    }
    for (Character character2 : string2.toCharArray()) {  //拆解为向量
        if (vectorMap.containsKey(character2)) {
            vectorMap.get(character2)[1]++;
        } else {
            tempArray = new int[2];
            tempArray[0] = 0;
            tempArray[1] = 1;  //第二个字符串的字符出现次数存储在数组的第二个位置
            vectorMap.put(character2, tempArray);
        }
    }
    /*
     * 到最后,map里的key对应的数组,第一个位置存储的是第一个字符串字符对应的向量,第二个位置存储的是第二个字符串字符对应的向量。
     */
}

// 求余弦相似度
public double sim() {   //分子分母相除
    double result = 0;
    result = pointMulti(vectorMap) / sqrtMulti(vectorMap);  
    return result;    //余弦度结果返回,因为是余弦,返回结果越大,夹角越小,两个向量方向越接近,即两个字符串越相似
}

//求平方根
private double sqrtMulti(Map paramMap) {  
    double result = 0;
    result = squares(paramMap);  //先求平方和
    result = Math.sqrt(result);  //再开根号,就是求模
    return result;
}

// 求平方和,分母上,向量求模的平方
private double squares(Map paramMap) {
    double result1 = 0;
    double result2 = 0;
    Set keySet = paramMap.keySet();
    for (Character character : keySet) {
        int temp[] = paramMap.get(character);  //获取key对应的值--数组
        result1 += (temp[0] * temp[0]);   //temp[0]存储的是第一个字符串对应的向量
        result2 += (temp[1] * temp[1]);   //temp[1]存储的是第二个字符串对应的向量
    }
    return result1 * result2;
}

// 点乘法,在分子上,向量相乘
private double pointMulti(Map paramMap) {
    double result = 0;
    Set keySet = paramMap.keySet();  //返回map中所有key值的列表,这里的set,也可以用list代替吧
    for (Character character : keySet) {    //存储的key值都是不重复的
        int temp[] = paramMap.get(character);   //获取key对应的值
        result += (temp[0] * temp[1]);
    }
    return resul 
    
}	

原理网上都有,这个代码也是网上的代码,我只是学习,加上了自己理解的注释。如有不足,请多指教。

你可能感兴趣的:(字符串相似度比较,余弦相似度算法,java实现余弦相似度,java)