Java实现的计算稀疏矩阵余弦相似度示例
本文实例讲述了Java实现的计算稀疏矩阵余弦相似度功能。分享给大家供大家参考，具体如下：
importjava.util.HashMap;
publicclassMyUDF{
/**
*UDFEvaluate接口
*
*UDF在记录层面上是一对一，字段上是一对一或多对一。Evaluate方法在每条记录上被调用一次，输入为一个或多个字段，输出为一个字段
*/
publicDoubleevaluate(Stringa,Stringb){
//TODO:请按需要修改参数和返回值，并在这里实现你自己的逻辑
if(a==null||b==null)
return0.0;
Stringtemp1[]=a.split(",");
Stringtemp2[]=b.split(",");
if(temp1==null||temp2==null){
return0.0;
}
HashMapmap1=newHashMap();
HashMapmap2=newHashMap();
for(Stringtemp:temp1)
{
Stringt[]=temp.split(":");
map1.put(t[0],Double.parseDouble(t[1]));
}
for(Stringtemp:temp2)
{
Stringt[]=temp.split(":");
map2.put(t[0],Double.parseDouble(t[1]));
}
doublefenzi=0;
doublefenmu1=0;
for(Stringi:map1.keySet())
{
doublevalue=map1.get(i);
if(map2.get(i)!=null){
fenzi+=value*map2.get(i);
}
fenmu1+=value*value;
}
doublefenmu2=0;
for(doublei:map2.values())
{
fenmu2+=i*i;
}
doublefenmu=Math.sqrt(fenmu1)*Math.sqrt(fenmu2);
returnfenzi/fenmu;
}
publicstaticvoidmain(String[]args){
Stringa="12:500,14:100,20:200";
Stringb="12:500,14:100,30:100";
MyUDFmyUDF=newMyUDF();
System.out.println(myUDF.evaluate(a,b));
}
}
运行结果：
0.9135468796041984
更多关于java算法相关内容感兴趣的读者可查看本站专题：《Java数据结构与算法教程》、《Java操作DOM节点技巧总结》、《Java文件与目录操作技巧汇总》和《Java缓存操作技巧汇总》
希望本文所述对大家java程序设计有所帮助。