Java實現的計算稀疏矩陣余弦相似度示例
更新時間:2018年07月09日 15:09:30 作者:Bryan__
這篇文章主要介紹了Java實現的計算稀疏矩陣余弦相似度功能,涉及java基于HashMap的數值計算相關操作技巧,需要的朋友可以參考下
本文實例講述了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節(jié)點技巧總結》、《Java文件與目錄操作技巧匯總》和《Java緩存操作技巧匯總》
希望本文所述對大家java程序設計有所幫助。
您可能感興趣的文章:
相關文章
java線程之使用Runnable接口創(chuàng)建線程的方法
本篇文章介紹了,java中使用Runnable接口創(chuàng)建線程的方法。需要的朋友參考下2013-05-05

