sklearn.metrics.pairwise_distances_argmin?
sklearn.metrics.pairwise_distances_argmin(X, Y, *, axis=1, metric='euclidean', metric_kwargs=None)
計算一個點與一組點之間的最小距離。
此函數為X中的每一行計算最接近(根據指定距離)的Y行的索引。
這主要等同于調用:
pairwise_distances(X,Y = Y,metric = metric).argmin(axis=axis)
但是使用更少的內存,并且對于大型數組更快。
此功能僅適用于密集2D陣列。
參數 | 說明 |
---|---|
X | array-like 包含點的數組。形狀分別為(n_samples1,n_features)和(n_samples2,n_features) |
Y | array-like 包含點的數組。形狀分別為(n_samples1,n_features)和(n_samples2,n_features) |
axis | int, optional, default 1 沿其計算argmin和距離的軸。 |
metric | string or callable 用于距離計算的指標。可以使用scikit-learn或scipy.spatial.distance中的任何度量。 如果metric是可調用的函數,則會在每對實例(行)上調用它,并記錄結果值。可調用對象應將兩個數組作為輸入并返回一個值,指示它們之間的距離。這適用于Scipy的指標,但效率不如將指標名稱作為字符串傳遞。 不支持距離矩陣。 指標的有效值為: - 來自scikit-learn:[‘cityblock’, ‘cosine’, ‘euclidean’, ‘l1’, ‘l2’, ‘manhattan’] - 來自scipy.spatial.distance:[‘braycurtis’, ‘canberra’, ‘chebyshev’, ‘correlation’, ‘dice’, ‘hamming’, ‘jaccard’, ‘kulsinski’, ‘mahalanobis’, ‘minkowski’, ‘rogerstanimoto’, ‘russellrao’, ‘seuclidean’, ‘sokalmichener’, ‘sokalsneath’, ‘sqeuclidean’, ‘yule’] 有關這些指標的詳細信息,請參見scipy.spatial.distance的文檔。 |
metric_kwargs | dict 要傳遞給指定度量函數的關鍵字參數。 |
返回值 | 說明 |
---|---|
argmin | numpy.ndarray Y[argmin[i], :]是Y中最接近X[i, :]的行。 |
另見: