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, :]的行。

另見:

sklearn.metrics.pairwise_distances

sklearn.metrics.pairwise_distances_argmin_min

sklearn.metrics.pairwise_distances_argmin應用示例?