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

6 votes301 views3 answers
Python moving average pandas
Python
0 vote299 views1 answer
Date format flouytter
Python
9 votes294 views3 answers
How to join tables in python
Python
15 votes295 views2 answers
Python check if variable is string
Python
0 vote301 views2 answers
Concatenate the squares of numbers in python
Python
4 votes269 views3 answers
What is a character in python
Python
2 votes294 views2 answers
Print list of list line by line python
Python
0 vote277 views1 answer
Reason: "broken data stream when reading image file" in jupyter notebook
Python
0 vote315 views1 answer
Django versatileimagefield
Python
1 vote292 views6 answers
Itertools
Python