第14章 聚类方法 - k均值聚类 - 实现Hierarchical类

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

class Hierarchical:
    def __init__(self, n_clusters=2):
        self.n_clusters = n_clusters
    
    def fit(self, X):
        self.linkage_matrix = linkage(X, method='ward')
        return self
    
    def predict(self, X):
        from scipy.cluster.hierarchy import fcluster
        return fcluster(self.linkage_matrix, self.n_clusters, criterion='maxclust')

X = np.random.rand(10, 2)
h = Hierarchical(n_clusters=2)
h.fit(X)
print("聚类完成")