Python计算两个矩阵的成对距离

最新推荐文章于 2023-08-15 14:04:33 发布

原创最新推荐文章于 2023-08-15 14:04:33 发布 · 762 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#python #矩阵 #机器学习

杂记专栏收录该内容

18 篇文章

订阅专栏

文章展示了三种在Python中计算10维特征向量间成对距离的方法：使用scikit-learn的pairwise_distances，scipy.spatial的distance_matrix和cdist函数。这三种方法分别在400000个和45000个样本的随机生成数据集上进行操作。

import numpy as np
from sklearn.metrics import pairwise_distances
from scipy.spatial import distance_matrix
from scipy.spatial.distance import cdist

# 10-dimensional features
x = np.random.rand(400000).reshape((-1, 10))
y = np.random.rand(45000).reshape((-1, 10))

def option1():
    dists = pairwise_distances(x, y)

def option2():
    dists = distance_matrix(x, y)

def option3():
    dists = cdist(x, y)