java实现余弦相似度
时间: 2023-08-01 09:14:31 浏览: 84
余弦相似度算法实现
5星 · 资源好评率100%
以下是Java实现余弦相似度的示例代码:
```
public class CosineSimilarity {
public static double cosineSimilarity(double[] vectorA, double[] vectorB) {
double dotProduct = 0.0;
double normA = 0.0;
double normB = 0.0;
for (int i = 0; i < vectorA.length; i++) {
dotProduct += vectorA[i] * vectorB[i];
normA += Math.pow(vectorA[i], 2);
normB += Math.pow(vectorB[i], 2);
}
return dotProduct / (Math.sqrt(normA) * Math.sqrt(normB));
}
public static void main(String[] args) {
double[] vectorA = {1.0, 2.0, 3.0};
double[] vectorB = {4.0, 5.0, 6.0};
double cosineSimilarity = cosineSimilarity(vectorA, vectorB);
System.out.println("Cosine similarity: " + cosineSimilarity);
}
}
```
其中`cosineSimilarity`方法接受两个向量作为参数,并返回两个向量的余弦相似度。在`main`方法中,我们定义了两个向量并计算它们的余弦相似度。
阅读全文