例如，如果我有一個 1D array[91, 119, 161, 203, 259]和一個 2D array [[90,120,160,200,260], [95,115,165,204,255]]，我如何確定后者的哪一行更適合前者？在這種情況下，它將是第一行，因為大多數數字僅相差 1，而在第二行中，大多數數字相差 4。此外，向量與第一行之間的 RMSE 為 1.612，而對于第二行它是 3.606。
1 回答
尚方寶劍之說
TA貢獻1788條經驗 獲得超4個贊
距離度量有很多，但余弦是衡量向量相似性的一個很好的方法。您可以用來scipy.spatial.distance.cosine求余弦距離。您需要具有最小余弦距離的向量。
在代碼中：
import numpy as np
from scipy.spatial.distance import cosine
v = np.array([91, 119, 161, 203, 259])
matrix = np.array([[90,120,160,200,260], [95,115,165,204,255]])
assert np.argmin([cosine(v, row) for row in matrix]) == 0
添加回答
舉報
0/150
提交
取消