余弦相似性代码

这是一个简单的Python代码示例，用于计算两个向量之间的余弦相似度：

import numpy as np

def cosine_similarity(a, b):
    dot_product = np.dot(a, b)
    norm_a = np.linalg.norm(a)
    norm_b = np.linalg.norm(b)
    return dot_product / (norm_a * norm_b)

# 示例
a = np.array([1, 2, 3])
b = np.array([4, 5, 6])
similarity = cosine_similarity(a, b)
print(similarity)

在这个示例中，我们使用NumPy库中的dot和linalg.norm函数来计算向量之间的点积和范数。然后，我们将点积除以这两个向量的范数的乘积，得到它们之间的余弦相似度。