agl.py

import numpy as np
import matplotlib.pyplot as plt
from sklearn.cluster import AgglomerativeClustering
from scipy.cluster.hierarchy import dendrogram, linkage

# Height and Weight of students
x = np.array([[150, 50],[152, 52],[155, 54],[170, 70],[172, 72],[175, 75]])

# linked = linkage(x, method='ward')
# plt.figure(figsize=(6,4))
# plt.subplot(1,2,1)
# dendrogram(linked)
# plt.title("Dendrogram")
# plt.xlabel("Students")
# plt.ylabel("Distance")

model = AgglomerativeClustering(n_clusters=2, linkage='ward')
labels = model.fit_predict(x)
print("Cluster Labels:", labels)

# plt.subplot(1,2,2)
plt.scatter(x[:,0], x[:,1], c=labels)
plt.xlabel("Height")
plt.ylabel("Weight")
plt.title("Agglomerative Clustering")
plt.show()