Download as txt, pdf, or txt
Download as txt, pdf, or txt
You are on page 1of 1

from sklearn.

cluster import AgglomerativeClustering


import numpy as np
import sklearn.datasets as datasets
from sklearn.model_selection import train_test_split
from sklearn.cluster import KMeans
from sklearn.cluster import AgglomerativeClustering
from sklearn.cluster import AffinityPropagation
from sklearn.metrics import homogeneity_score
iris = datasets.load_iris()
X = iris.data
km_cls = KMeans(n_clusters=3, ).fit(X)
y_pred = km_cls.predict(X)
print(homogeneity_score(iris.target,y_pred))
agg_cls = AgglomerativeClustering(n_clusters=3)
# agg_cls = agg_cls.fit(X)
y_pred1 = agg_cls.fit_predict(X)
print(homogeneity_score(iris.target,y_pred1))

af_cls = AffinityPropagation().fit(X)
af_cls
y_pred2 = af_cls.predict(X)
print(homogeneity_score(iris.target,y_pred2))

You might also like