23 lines
673 B
Python
23 lines
673 B
Python
import numpy as np
|
|
|
|
feature_map1 = np.array([[1, 0, 1], [0, 1, 0], [1, 0, 1]])
|
|
feature_map2 = np.array([[1, 1, 1], [1, 0, 0], [1, 0, 0]])
|
|
|
|
vector1 = feature_map1.flatten()
|
|
vector2 = feature_map2.flatten()
|
|
|
|
print("vector1:", vector1)
|
|
print("vector2:", vector2)
|
|
print()
|
|
|
|
euclidean_distance = np.linalg.norm(vector1 - vector2)
|
|
print(f"欧几里得距离: {euclidean_distance:.4f}")
|
|
|
|
dot_product = np.dot(vector1, vector2)
|
|
norm1 = np.linalg.norm(vector1)
|
|
norm2 = np.linalg.norm(vector2)
|
|
cosine_similarity = dot_product / (norm1 * norm2)
|
|
print(f"余弦相似度: {cosine_similarity:.4f}")
|
|
|
|
cosine_distance = 1 - cosine_similarity
|
|
print(f"余弦距离: {cosine_distance:.4f}") |