from sklearn.cluster import AgglomerativeClustering
# Set distance_threshold instead of n_clusters to let the algorithm determine clusters
hc = AgglomerativeClustering(distance_threshold=1.0, n_clusters=None)
cluster_labels = hc.fit_predict(embeddings)
Editor is loading...
