pairwise_distances_argmin_min

sklearn.metrics.pairwise_distances_argmin_min(X, Y, *, axis=1, metric='euclidean', metric_kwargs=None)[source]

Вычисляет минимальные расстояния между одной точкой и набором точек.

Для каждой строки X эта функция вычисляет индекс ближайшей строки Y (согласно указанному расстоянию). Также возвращаются минимальные расстояния.

В основном это эквивалентно вызову:

(pairwise_distances(X, Y=Y, metric=metric).argmin(axis=axis),
 pairwise_distances(X, Y=Y, metric=metric).min(axis=axis))

но требует значительно меньше памяти и работает быстрее с большими массивами.

Параметры:

X{array-like, sparse matrix} of shape (n_samples_X, n_features)

Массив, содержащий точки.

Y{array-like, sparse matrix} of shape (n_samples_Y, n_features)

Массив, содержащий точки.

axisint, default=1

Ось, вдоль которой вычисляются argmin и расстояния.

metricstr or callable, default=’euclidean’

Метрика для вычисления расстояния. Можно использовать любую метрику из scikit-learn или scipy.spatial.distance.

Если metric — вызываемая функция, она вызывается для каждой пары экземпляров (строк), а полученное значение сохраняется. Функция должна принимать на вход два массива и возвращать одно значение, указывающее расстояние между ними. Это работает с метриками SciPy, но менее эффективно, чем передача имени метрики в виде строки.

Матрицы расстояний не поддерживаются.

Допустимые значения metric:

• из scikit-learn: [‘cityblock’, ‘cosine’, ‘euclidean’, ‘l1’, ‘l2’, ‘manhattan’, ‘nan_euclidean’]
• из scipy.spatial.distance: [‘braycurtis’, ‘canberra’, ‘chebyshev’, ‘correlation’, ‘dice’, ‘hamming’, ‘jaccard’, ‘kulsinski’, ‘mahalanobis’, ‘minkowski’, ‘rogerstanimoto’, ‘russellrao’, ‘seuclidean’, ‘sokalmichener’, ‘sokalsneath’, ‘sqeuclidean’, ‘yule’]

Подробные сведения об этих метриках см. в документации scipy.spatial.distance.

Примечание

'kulsinski' объявлена устаревшей в SciPy 1.9 и будет удалена в SciPy 1.11.

Примечание

'matching' удалена в SciPy 1.9 (вместо неё используйте 'hamming').

metric_kwargsdict, default=None

Именованные аргументы, передаваемые указанной функции метрики.

Возвращает:

argminndarray

Y[argmin[i], :] — строка Y, ближайшая к X[i, :].

distancesndarray

Массив минимальных расстояний. distances[i] — это расстояние между i-й строкой X и argmin[i]-й строкой Y.

См. также

pairwise_distances

Расстояния между каждой парой образцов из X и Y.

pairwise_distances_argmin

То же, что pairwise_distances_argmin_min, но возвращает только argmin.

Примеры

>>> from sklearn.metrics.pairwise import pairwise_distances_argmin_min
>>> X = [[0, 0, 0], [1, 1, 1]]
>>> Y = [[1, 0, 0], [1, 1, 0]]
>>> argmin, distances = pairwise_distances_argmin_min(X, Y)
>>> argmin
array([0, 1])
>>> distances
array([1., 1.])