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.

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

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

Допустимые значения для метрики:

• из 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.])