from sklearn.cluster import MeanShift, estimate_bandwidth bandwidth = estimate_bandwidth(embeddings, quantile=0.2) ms = MeanShift(bandwidth=bandwidth, bin_seeding=True) cluster_labels = ms.fit_predict(embeddings)
