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 (); HashMap map2=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程序设计有所帮助。