欢迎访问宙启技术站
智能推送

基于BernoulliNB()算法的Python文本分类示例

发布时间:2023-12-11 08:08:26

文本分类是一种将文本分为不同类别的任务,例如将电子邮件分类为垃圾邮件或非垃圾邮件,将新闻文章分类为政治、体育或娱乐等。Python提供了许多用于文本分类的库和算法,其中之一是基于BernoulliNB()算法的朴素贝叶斯分类器。

朴素贝叶斯分类器是一种基于贝叶斯定理和特征条件独立性的分类算法。BernoulliNB()是一种朴素贝叶斯分类器的变体,适用于二分类问题。它假设每个特征是二进制的,即每个特征的取值只能为0或1。该算法使用每个特征在类别中的出现频率来计算属于每个类别的概率。

下面是一个使用BernoulliNB()算法进行文本分类的示例:

# 导入所需的库和模块
from sklearn.feature_extraction.text import CountVectorizer
from sklearn.naive_bayes import BernoulliNB

# 定义训练集和测试集
train_texts = ["I love this movie", "This movie is great", "I hate this movie"]
train_labels = [1, 1, 0]
test_texts = ["This movie is terrible", "I like this movie"]

# 创建特征向量
vectorizer = CountVectorizer(binary=True)
train_features = vectorizer.fit_transform(train_texts)
test_features = vectorizer.transform(test_texts)

# 创建BernoulliNB分类器并进行训练
classifier = BernoulliNB()
classifier.fit(train_features, train_labels)

# 进行分类
predictions = classifier.predict(test_features)
print(predictions)

在上面的示例中,首先导入了所需的库和模块。然后定义了训练集和测试集,其中训练集包括了几个句子和对应的类别标签。接下来,使用CountVectorizer将文本转换为特征向量,该向量表示每个单词在句子中是否出现。然后,使用BernoulliNB()创建了一个分类器,并通过调用fit方法进行训练。最后,使用predict方法对测试集进行分类,并将结果打印出来。

在这个例子中,训练集中有3个句子,其中2个被标记为正面评价(1),1个被标记为负面评价(0)。测试集中有2个句子,根据它们的特征向量被分类为正面还是负面评价。

需要注意的是,这个示例只用来演示BernoulliNB()算法的用法,实际应用中可以使用更大的训练集和测试集,并根据需要进行特征工程和模型调参等步骤来提高分类的准确性。

总结起来,基于BernoulliNB()算法的Python文本分类示例可以通过使用sklearn库中的CountVectorizer和BernoulliNB来构建特征向量和分类器,并通过fit和predict等方法进行训练和分类。