gpt4 book ai didi

python-3.x - 如何获得离簇中心最近的N个数据点?

转载 作者:行者123 更新时间:2023-11-30 08:34:38 25 4
gpt4 key购买 nike

在部署 K-means 算法后,我想在每个集群中获取距中心最近的 N 个数据点(基于欧几里得距离)。我可以使用获取数据点的索引

np.where(km.labels_ == 0)

最佳答案

您可以使用 kmeans 类的 transform 方法来计算每个数据点到每个簇的距离。

然后假设您想要第 0 个索引中的前 N 个点,那么您可以这样做:

cluster = 0
N = 2
np.sort(kmeans.transform(X)[:,cluster])[:N]

关于python-3.x - 如何获得离簇中心最近的N个数据点?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/53790832/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com