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 vote225 views1 answer
Seaborn angle lable
Python
0 vote219 views1 answer
Real hour in python
Python
12 votes275 views4 answers
Flask debug
Python Flask
7 votes244 views5 answers
How to multiply two lists in python
Python
0 vote224 views1 answer
Python convert images to pdf
Python
8 votes195 views4 answers
Reverse dictionary python
Python
1 vote215 views1 answer
Knowing the sum null values in a specific row in pandas dataframe
Python
7 votes234 views3 answers
Python argparse choice
Python
0 vote238 views1 answer
Python unzip list of tuples
Python
0 vote257 views1 answer
Matplotlib different number of subplots
Python