Java實現的計算稀疏矩陣餘弦相似度示例

本文實例講述了Java實現的計算稀疏矩陣餘弦相似度功能。分享給大家供大家參考，具體如下：

 import java.util.HashMap;public class MyUDF{ /** * UDF Evaluate接口* * UDF在記錄層面上是一對一，字段上是一對一或多對一。 Evaluate方法在每條記錄上被調用一次，輸入為一個或多個字段，輸出為一個字段*/ public Double evaluate(String a, String b) { // TODO: 請按需要修改參數和返回值，並在這裡實現你自己的邏輯if(a==null || b==null) return 0.0; String temp1[]=a.split(","); String temp2[]=b.split(","); if (temp1==null || temp2==null) { return 0.0; } HashMap<String, Double> map1=new HashMap<String, Double>(); HashMap<String, Double> map2=new HashMap<String, Double>(); for(String temp:temp1) { String t[]=temp.split(":"); map1.put(t[0], Double.parseDouble(t[1])); } for(String temp:temp2) { String t[]=temp.split(":"); map2.put(t[0], Double.parseDouble(t[1])); } double fenzi=0; double fenmu1=0; for(String i:map1.keySet()) { double value=map1.get(i); if (map2.get(i)!=null) { fenzi+=value*map2.get(i); } fenmu1+=value*value; } double fenmu2=0; for(double i:map2.values()) { fenmu2+=i*i; } double fenmu=Math.sqrt(fenmu1)*Math.sqrt(fenmu2); return fenzi/fenmu; } public static void main(String[] args) { String a="12:500,14:100,20:200"; String b="12:500,14:100,30:100"; MyUDF myUDF=new MyUDF(); System.out.println(myUDF.evaluate(a, b)); }}

運行結果：

0.9135468796041984

更多關於java算法相關內容感興趣的讀者可查看本站專題：《Java數據結構與算法教程》、《Java操作DOM節點技巧總結》、《Java文件與目錄操作技巧匯總》和《Java緩存操作技巧匯總》

希望本文所述對大家java程序設計有所幫助。