package recommendAlgo;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
public class testRecommend {
public static void main(String[] args) {
Map<String, Map<String, Integer>> userPerfMap = new HashMap<String, Map<String, Integer>>();
Map<String, Integer> pref1 = new HashMap<String, Integer>();
pref1.put("A", 3);
pref1.put("B", 4);
pref1.put("C", 3);
pref1.put("D", 5);
pref1.put("E", 1);
pref1.put("F", 4);
userPerfMap.put("p1", pref1);
Map<String, Integer> pref2 = new HashMap<String, Integer>();
pref2.put("A", 2);
pref2.put("B", 4);
pref2.put("C", 4);
pref2.put("D", 5);
pref2.put("E", 3);
pref2.put("F", 2);
userPerfMap.put("p2", pref2);
Map<String, Integer> pref3 = new HashMap<String, Integer>();
pref3.put("A", 3);
pref3.put("B", 5);
pref3.put("C", 4);
pref3.put("D", 5);
pref3.put("E", 2);
pref3.put("F", 1);
userPerfMap.put("p3", pref3);
Map<String, Integer> pref4 = new HashMap<String, Integer>();
pref4.put("A", 2);
pref4.put("B", 2);
pref4.put("C", 3);
pref4.put("D", 4);
pref4.put("E", 3);
pref4.put("F", 2);
userPerfMap.put("p4", pref4);
Map<String, Integer> pref5 = new HashMap<String, Integer>();
pref5.put("A", 4);
pref5.put("B", 4);
pref5.put("C", 4);
pref5.put("D", 5);
pref5.put("E", 1);
pref5.put("F", 0);
userPerfMap.put("p5", pref5);
Map<String, Double> simUserSimMap = new HashMap<String, Double>();
String output1 = "皮尔逊相关系数:", output2 = "欧几里得距离:";
for (Entry<String, Map<String, Integer>> userPerfEn : userPerfMap.entrySet()) {
String userName = userPerfEn.getKey();
if (!"p5".equals(userName)) {
double sim = getUserSimilar(pref5, userPerfEn.getValue());
double distance = getEuclidDistance(pref5, userPerfEn.getValue());
output1 += "p5与" + userName + "之间的相关系数:" + sim + ",";
output2 += "p5与" + userName + "之间的距离:" + distance + ",";
simUserSimMap.put(userName, sim);
}
}
System.out.println(output1);
System.out.println(output2);
Map<String, Map<String, Integer>> simUserObjMap = new HashMap<String, Map<String, Integer>>();
Map<String, Integer> pobjMap1 = new HashMap<String, Integer>();
pobjMap1.put("一夜惊喜", 3);
pobjMap1.put("环太平洋", 4);
pobjMap1.put("变形金刚", 3);
simUserObjMap.put("p1", pobjMap1);
Map<String, Integer> pobjMap2 = new HashMap<String, Integer>();
pobjMap2.put("一夜惊喜", 5);
pobjMap2.put("环太平洋", 1);
pobjMap2.put("变形金刚", 2);
simUserObjMap.put("p2", pobjMap2);
Map<String, Integer> pobjMap3 = new HashMap<String, Integer>();
pobjMap3.put("一夜惊喜", 2);
pobjMap3.put("环太平洋", 5);
pobjMap3.put("变形金刚", 5);
simUserObjMap.put("p3", pobjMap3);
System.out.println("根据系数推荐:" + getRecommend(simUserObjMap, simUserSimMap));
}
/**
*
* @Title getUserSimilar
* @Class testRecommend
* @return double
* @param pm1
* @param pm2
* @return
* @Description获取两个用户之间的皮尔逊相似度,相关系数的绝对值越大,相关度越大
* @author qinshijiang
* @Date 2013-9-4
*/
public static double getUserSimilar(Map<String, Integer> pm1, Map<String, Integer> pm2) {
int n = 0;// 数量n
int sxy = 0;// Σxy=x1*y1+x2*y2+....xn*yn
int sx = 0;// Σx=x1+x2+....xn
int sy = 0;// Σy=y1+y2+...yn
int sx2 = 0;// Σx2=(x1)2+(x2)2+....(xn)2
int sy2 = 0;// Σy2=(y1)2+(y2)2+....(yn)2
for (Entry<String, Integer> pme : pm1.entrySet()) {
String key = pme.getKey();
Integer x = pme.getValue();
Integer y = pm2.get(key);
if (x != null && y != null) {
n++;
sxy += x * y;
sx += x;
sy += y;
sx2 += Math.pow(x, 2);
sy2 += Math.pow(y, 2);
}
}
// p=(Σxy-Σx*Σy/n)/Math.sqrt((Σx2-(Σx)2/n)(Σy2-(Σy)2/n));
double sd = sxy - sx * sy / n;
double sm = Math.sqrt((sx2 - Math.pow(sx, 2) / n) * (sy2 - Math.pow(sy, 2) / n));
return Math.abs(sm == 0 ? 1 : sd / sm);
}
/**
*
* @Title getEuclidDistance
* @Class testRecommend
* @return double
* @param pm1
* @param pm2
* @return
* @Description获取两个用户之间的欧几里得距离,距离越小越好
* @author qinshijiang
* @Date 2013-9-4
*/
public static double getEuclidDistance(Map<String, Integer> pm1, Map<String, Integer> pm2) {
double totalscore = 0.0;
for (Entry<String, Integer> test : pm1.entrySet()) {
String key = test.getKey();
Integer a1 = pm1.get(key);
Integer b1 = pm2.get(key);
if (a1 != null && b1 != null) {
double a = Math.pow(a1 - b1, 2);
totalscore += Math.abs(a);
}
}
return Math.sqrt(totalscore);
}
/**
*
* @Title getRecommend
* @Class testRecommend
* @return String
* @param simUserObjMap
* @param simUserSimMap
* @return
* @Description根据相关系数得到推荐物品
* @author qinshijiang
* @Date 2013-9-4
*/
public static String getRecommend(Map<String, Map<String, Integer>> simUserObjMap, Map<String, Double> simUserSimMap) {
Map<String, Double> objScoreMap = new HashMap<String, Double>();
for (Entry<String, Map<String, Integer>> simUserEn : simUserObjMap.entrySet()) {
String user = simUserEn.getKey();
double sim = simUserSimMap.get(user);
for (Entry<String, Integer> simObjEn : simUserEn.getValue().entrySet()) {
double objScore = sim * simObjEn.getValue();
String objName = simObjEn.getKey();
if (objScoreMap.get(objName) == null) {
objScoreMap.put(objName, objScore);
}else {
double totalScore = objScoreMap.get(objName);
objScoreMap.put(objName, totalScore + objScore);
}
}
}
List<Entry<String, Double>> enList = new ArrayList<Entry<String, Double>>(objScoreMap.entrySet());
Collections.sort(enList, new Comparator<Map.Entry<String, Double>>() {
public int compare(Map.Entry<String, Double> o1, Map.Entry<String, Double> o2) {
Double a = o1.getValue() - o2.getValue();
if (a == 0) {
return 0;
}else if (a > 0) {
return 1;
}else {
return -1;
}
}
});
return enList.get(enList.size() - 1).getKey();
}
}