怎么使用TextBlob进行交叉验证

TextBlob

小亿

2024-05-13 12:17:15

栏目: 编程语言

导入必要的库和数据集：

from textblob import TextBlob
from sklearn.model_selection import cross_val_score
from sklearn.feature_extraction.text import CountVectorizer
from sklearn.naive_bayes import MultinomialNB
from sklearn.pipeline import make_pipeline
from sklearn.datasets import fetch_20newsgroups

加载数据集：

categories = ['alt.atheism', 'comp.graphics', 'sci.med', 'soc.religion.christian']
data = fetch_20newsgroups(categories=categories)
X = data.data
y = data.target

创建pipeline，包括文本向量化和分类模型：

model = make_pipeline(CountVectorizer(), MultinomialNB())

使用cross_val_score进行交叉验证：

scores = cross_val_score(model, X, y, cv=5, scoring='accuracy')
print("Cross-validation scores: ", scores)
print("Average score: ", scores.mean())

这样，你就可以使用TextBlob进行交叉验证了。

怎么使用TextBlob进行交叉验证

最新问答

相关标签