要在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ù)。