sklearn 学习 Mean Shift聚类

 import numpy as np

    
 from sklearn.cluster import MeanShift, estimate_bandwidth
    
 from sklearn.datasets.samples_generator import make_blobs
    
  
    
 # Generate sample data 创建数据集
    
 centers = [[1, 1], [-1, -1], [1, -1]]
    
 X, _ = make_blobs(n_samples=10000, centers=centers, cluster_std=0.6)
    
  
    
 bandwidth = estimate_bandwidth(X, quantile=0.2, n_samples=500)
    
  
    
 ms = MeanShift(bandwidth=bandwidth, bin_seeding=True)
    
 ms.fit(X)
    
 labels = ms.labels_
    
 cluster_centers = ms.cluster_centers_
    
  
    
 labels_unique = np.unique(labels)