親和性中的相似性矩陣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_