cosine_similarity

sklearn.metrics.pairwise.cosine_similarity(X, Y=None, dense_output=True)[исходный код]

Вычисляет косинусное сходство между объектами в X и Y.

Косинусное сходство, или косинусное ядро, вычисляет сходство как нормализованное скалярное произведение X и Y:

K(X, Y) = <X, Y> / (||X||*||Y||)

Для данных, нормализованных по L2-норме, эта функция эквивалентна linear_kernel.

Подробнее см. в Руководстве пользователя.

Параметры:

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

Входные данные.

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

Входные данные. Если None, выходные данные будут содержать попарные сходства между всеми объектами в X.

dense_outputbool, default=True

Возвращать ли плотный результат, даже если входные данные разреженные. Если False, результат будет разреженным, если оба входных массива разреженные.

Добавлено в версии 0.17: параметр dense_output для плотного результата.

Возвращает:

similaritiesndarray or sparse matrix of shape (n_samples_X, n_samples_Y)

Возвращает косинусное сходство между объектами в X и Y.

Примеры

>>> from sklearn.metrics.pairwise import cosine_similarity
>>> X = [[0, 0, 0], [1, 1, 1]]
>>> Y = [[1, 0, 0], [1, 1, 0]]
>>> cosine_similarity(X, Y)
array([[0.   , 0.   ],
       [0.577, 0.816]])

Примеры из галереи

Построение границ классификации с различными ядрами SVM