Python亲和传播中的相似矩阵

Question

我想在python中运行相似性传播。我知道sklearn.cluster公司有这个吗算法好像是这样sklearn.cluster.AffinityPropagation通过将相似矩阵计算为负欧几里德距离来运行它。但是，我有自己的相似矩阵，不想使用欧几里德距离。 所以我想知道有没有办法把我的相似度矩阵作为输入，然后运行基于它的算法？在

Answer 1

是的，你能做到的！请参阅以下片段

from sklearn.cluster import AffinityPropagation

# compute the similarity matrix
af = AffinityPropagation(affinity='precomputed', verbose=True)
af.fit(similarity_matrix)
print af.labels_, af.cluster_centers_indices_