Affinity propagation python

from sklearn.cluster import AffinityPropagation
import numpy as np

X = np.array([[1, 2], [1, 4], [1, 0], [4, 2], [4, 4], [4, 0]])
clustering = AffinityPropagation(affinity = 'euclidean', random_state=5).fit(X)

labels = clustering.labels_ # label to each element
centers = clustering.cluster_centers_ # center of each cluster

# if you need a distance different from euclidean
# calculate your custom, pairwise distance among vectors 
# and store them into a matrix M. 
# Note: cluster_centers are no longer available

clustering = AffinityPropagation(affinity='precomputed', random_state=5).fit(M)

Source: scikit-learn.org

Similar codes

0 vote185 views1 answer
Python csv reader cast to float
Python
0 vote203 views1 answer
Concatenate dataframes using one column
Python
0 vote201 views1 answer
Python setup install_requires local whl
Python
4 votes203 views1 answer
Python zip file open as text
Python
0 vote190 views1 answer
Size of the query process in python BigQuery
Python
28 votes203 views2 answers
Get a slice of string in python
Python
19 votes191 views5 answers
Pandas where retuning NaN
Python
1 vote190 views1 answer
Les librairies python a maitriser pour faire du machine learning
Python
2 votes208 views1 answer
Python Key–value database
Python
0 vote204 views1 answer
How to add import pydictionary in python
Python