余弦相似度简单介绍

public class CosineSimilarityCalculator {

/**
* 计算两个向量的余弦相似度
*
* @param vectorA 第一个向量
* @param vectorB 第二个向量
* @return 余弦相似度
* @throws IllegalArgumentException 如果两个向量的长度不同
*/
public static double cosineSimilarity(double[] vectorA, double[] vectorB) {
if (vectorA.length != vectorB.length) {
throw new IllegalArgumentException("两个向量的长度必须相同");
}

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);
}

// 计算模长
double magnitudeA = Math.sqrt(normA);
double magnitudeB = Math.sqrt(normB);

// 防止除以0
if (magnitudeA == 0 || magnitudeB == 0) {
return 0.0; // 如果其中一个向量为零向量，则余弦相似度为0
}

// 计算余弦相似度
return dotProduct / (magnitudeA * magnitudeB);
}

public static void main(String[] args) {
double[] vectorA = {5, 3, 0, 1, 4};
double[] vectorB = {4, 2, 1, 0, 5};

try {
double similarity = cosineSimilarity(vectorA, vectorB);
System.out.println("Cosine Similarity: " + similarity);
} catch (IllegalArgumentException e) {
System.out.println(e.getMessage());
}
}
}