spaCy怎么可視化文本聚類

小億
88
2024-05-11 17:51:56

要在spaCy中可視化文本聚類,通??梢允褂胹cattertext庫(kù)。以下是一個(gè)示例代碼,展示了如何在spaCy中使用scattertext庫(kù)可視化文本聚類:

import spacy
import scattertext as st
from scattertext import CorpusFromPandas

# 加載spaCy模型
nlp = spacy.load("en_core_web_sm")

# 創(chuàng)建一個(gè)示例數(shù)據(jù)集
data = {
    'text': ['This is a great product', 'I love this product', 'Not satisfied with this product', 'Highly recommended', 'Waste of money'],
    'category': ['positive', 'positive', 'negative', 'positive', 'negative']
}

df = pd.DataFrame(data)

# 使用spaCy進(jìn)行處理文本
corpus = st.CorpusFromPandas(df, category_col='category', text_col='text', nlp=nlp).build()

# 創(chuàng)建一個(gè)可視化
html = st.produce_scattertext_explorer(corpus, category='positive', category_name='Positive', not_category_name='Negative')

# 保存可視化為HTML文件
open('visualization.html', 'w').write(html)

在這個(gè)示例中,我們首先加載了spaCy模型,然后創(chuàng)建了一個(gè)包含文本和類別的示例數(shù)據(jù)集。接下來(lái),我們使用spaCy對(duì)文本進(jìn)行處理,并使用scattertext庫(kù)構(gòu)建了語(yǔ)料庫(kù)。最后,我們使用produce_scattertext_explorer函數(shù)創(chuàng)建了一個(gè)可視化,并將其保存為HTML文件。

您可以根據(jù)您的需要調(diào)整數(shù)據(jù)集和可視化選項(xiàng),以適應(yīng)不同的文本聚類任務(wù)。

0