Question

我正在做一些尝试推荐文档的工作，为此我使用了Cosine Similarity方法。以下是该方法的代码：

static double cosineSimilarity(HashMap<String, Double> v1, HashMap<String, Double> v2) 
{
    Set<String> both = v1.keySet();
    both.retainAll(v2.keySet());
    double sclar = 0, norm1 = 0, norm2 = 0;
    for (String k : both) 
    {
      sclar += v1.get(k) * v2.get(k);
    }
    for (String k : v1.keySet())
    {
      norm1 += v1.get(k) * v1.get(k);
    }
    for (String k : v2.keySet()) 
    {
      norm2 += v2.get(k) * v2.get(k);
    }
    return sclar / Math.sqrt(norm1 * norm2);
}

问题是结果会根据参数的传递顺序而有所不同。例如，如果我致电cosineSimilarity(v1, v2)，它将返回0.3，但如果我致电cosineSimilarity(v2, v1)，则会返回完全不同的值。

我认为这与Map.keySet()返回地图支持的集合这一事实有关，但我并不完全理解这一点的含义。

任何人都可以看到方法出错的地方吗？

Answer 1

尝试

Set<String> both = new HashSet<String>(v1.keySet());

而不是

Set<String> both = v1.keySet();

您不应该修改从keySet方法获得的集合，因为地图使用它，当您从中删除元素时（在您的情况下由retainAll删除）元素也会从地图中删除。例如：

Map<Integer, Integer> mp = new HashMap<Integer, Integer>();
mp.put(1, 1);
System.out.println(mp); // output {1=1}
mp.keySet().remove(1);
System.out.println(mp); // output {}

在Java中使用Maps时出现意外/未定义的结果

1 个答案: