sklearn中的凝聚聚类

时间:2017-06-29 21:02:16

标签: scikit-learn hierarchical-clustering

我有一些数据以及这些数据点的成对距离矩阵。我想使用Agglomerative clustering对它们进行聚类。我在sklearn中读到,我们可以预先计算出'作为亲和力,我希望它是距离矩阵。但我找不到任何使用预计算亲和力和自定义距离矩阵的示例。 任何帮助将不胜感激。

1 个答案:

答案 0 :(得分:11)

让我们调用您的距离矩阵D

agg = AgglomerativeClustering(n_clusters=5, affinity='precomputed')
agg.fit_predict(D)  # Returns class labels.

如果您对生成整个层次结构并生成树形图感兴趣,scikit-learn的API将包裹scipy hierarchical clustering code。只需直接使用scipy代码。