0

我目前正在尝试使用 HDBSCAN 对一堆电影数据进行聚类，以便将相似的内容组合在一起并能够提出描述这些聚类的“主题”。我对 HDBSCAN 很感兴趣，因为我知道它被认为是软聚类，而不是 K-Means，它更适合我的目标。

执行 HDBSCAN 后，我能够找到每个群集中放置的电影。我现在想要的是代表每个集群的术语/单词。

我用 KMeans 做了类似的事情（下面的代码）：

model = KMeans(n_clusters=70)
model.fit(text)
clusters=model.predict(text)
model_labels=model.labels_
output= model.transform(text)

titles=[]
for i in data['title']:
        titles.append(i)
genres=[]
for i in data['genres']:
        genres.append(i)

films_kmeans = { 'title': titles, 'info': dataset_list2, 'cluster': clusters, 'genre': genres }
frame_kmeans= pd.DataFrame(films_kmeans, index=[clusters])

print("Top terms per cluster:")
print()
#sort cluster centers by proximity to centroid
order_centroids = model.cluster_centers_.argsort()[:, ::-1] 
for i in range(70):
    print("Cluster %d:" % i),
    for ind in order_centroids[i, :5]:
        print(' %s' % tfidf_feature_names[ind]),
    print()
    print()

    print("Cluster %d titles:" % i, end='')
    for title in frame_kmeans.loc[i]['title'].values.tolist():
        print(' %s,' % title, end='')
    print() #add whitespace
    print() #add whitespace

print()

虽然这适用于 KMeans，但我无法为 HDBSCAN 找到类似的方法，因为我知道它没有集群中心。我一直在查看文档，但我对此很陌生，我无法解决我的问题。

任何想法将不胜感激！感谢您的时间。

4

2 回答 2

1

我遇到了类似的问题，并从@ajmartin 的建议中带头，下面的代码对我有用。假设您有一个标签列表 -label包含每个点的原始标签和一个 HDBSCAN 对象clusterer = hdbscan.HDBSCAN(min_cluster_size=10).fit(X)，

from operator import itemgetter
from collections import defaultdict

def get_top_terms(k):
    top_terms = defaultdict(list)
    for c_lab, prob, text_lab in zip(clusterer.labels_, clusterer.probabilities_, label):
        top_terms[c_lab].append((prob, text_lab))

    for c_lab in top_terms:
        top_terms[c_lab].sort(reverse=True, key=itemgetter(0)) # sort the pair based on probability 

    # -- print the top k terms per cluster --    
    for c_lab in top_terms:
        print(c_lab, top_terms[c_lab][:k])
    return top_terms

# -- for visualization (add this snippet before plt.scatter(..))--
from collections import Counter

plt.figure(figsize=(16, 16))
plt.title('min_cluster_size=10')

plot_top=Counter() # to get only distinct labels, replace with a set and add a check here [1] 
top_terms = get_top_terms(10)

for i, lab, prob in zip(range(len(clusterer.labels_)),clusterer.labels_, clusterer.probabilities_): # pointwise iteration
    if plot_top[lab] < 10:      
        for el in top_terms[lab][:10]:
            if prob == el[0]: # [1] 
                plot_top[lab] += 1
                # x[i], y[i] are the projected points in 2D space 
                plt.annotate(el[1], (x[i],y[i]), horizontalalignment='center', verticalalignment='center', size=9.5)
                break
于 2019-09-05T13:24:48.487 回答
0

参考 HDBSCAN教程。对于由算法聚类的每个样本，它还关联一个概率，该概率可以被认为是样本与聚类的关联程度。您可以过滤每个集群的样本及其相应的概率；使用概率来确定每个集群的最高点。该链接有更多详细信息。

于 2019-08-01T09:19:32.533 回答