用python写一个朴素贝叶斯算法实现
时间: 2023-02-17 11:08:45 浏览: 150
python实现朴素贝叶斯算法
5星 · 资源好评率100%
Python中使用朴素贝叶斯算法实现的示例代码如下:
```python
from sklearn.naive_bayes import MultinomialNB
from sklearn.feature_extraction.text import CountVectorizer
# 训练数据
train_data = ["这是一个好的文章", "这是一篇非常好的文章", "这是一篇很差的文章"]
train_label = [1, 1, 0] # 1表示好文章,0表示差文章
# 测试数据
test_data = ["这是一篇非常优秀的文章"]
# 将文本转换为特征向量
vec = CountVectorizer()
vec.fit(train_data)
train_vec = vec.transform(train_data)
test_vec = vec.transform(test_data)
# 训练模型
clf = MultinomialNB()
clf.fit(train_vec, train_label)
# 预测结果
prediction = clf.predict(test_vec)
print(prediction)
```
这段代码使用了sklearn库中的MultinomialNB类和CountVectorizer类,它们分别用于实现朴素贝叶斯算法和文本特征提取。
这段代码训练了一个分类器,并使用这个分类器对测试数据进行预测。
注意:朴素贝叶斯算法是基于概率的算法,本示例是二元分类问题,请根据实际问题进行调整.
阅读全文