python朴素贝叶斯
时间: 2023-08-19 07:11:04 浏览: 24
Python中的朴素贝叶斯算法可以通过sklearn库来实现。在sklearn中,有三种朴素贝叶斯算法的实现:伯努利朴素贝叶斯、高斯朴素贝叶斯和多项式朴素贝叶斯。伯努利朴素贝叶斯适用于二值型特征,高斯朴素贝叶斯适用于连续型特征,而多项式朴素贝叶斯适用于离散型特征。\[1\]
下面是使用sklearn库实现朴素贝叶斯算法的示例代码:
```python
# 导入所需库
from sklearn.naive_bayes import GaussianNB, BernoulliNB
# 实例化高斯朴素贝叶斯算法
gaussian_nb = GaussianNB()
# 实例化伯努利朴素贝叶斯算法
bernoulli_nb = BernoulliNB()
# 使用高斯朴素贝叶斯算法进行拟合和预测
gaussian_nb.fit(X_train, y_train)
y_pred_gaussian = gaussian_nb.predict(X_test)
# 使用伯努利朴素贝叶斯算法进行拟合和预测
bernoulli_nb.fit(X_train, y_train)
y_pred_bernoulli = bernoulli_nb.predict(X_test)
# 查看预测结果和预测概率
print('高斯朴素贝叶斯预测结果:', y_pred_gaussian)
print('高斯朴素贝叶斯预测概率:', gaussian_nb.predict_proba(X_test))
print('伯努利朴素贝叶斯预测结果:', y_pred_bernoulli)
print('伯努利朴素贝叶斯预测概率:', bernoulli_nb.predict_proba(X_test))
# 查看准确率
score_gaussian = gaussian_nb.score(X_test, y_test)
score_bernoulli = bernoulli_nb.score(X_test, y_test)
print('高斯朴素贝叶斯准确率:', score_gaussian)
print('伯努利朴素贝叶斯准确率:', score_bernoulli)
```
以上代码展示了如何使用sklearn库中的高斯朴素贝叶斯和伯努利朴素贝叶斯算法进行拟合、预测和评估准确率。你可以根据自己的数据类型选择适合的朴素贝叶斯算法进行分类任务。\[2\]\[3\]
#### 引用[.reference_title]
- *1* *2* *3* [Python机器学习--分类算法--朴素贝叶斯算法](https://blog.csdn.net/qq_43214331/article/details/120026248)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^control_2,239^v3^insert_chatgpt"}} ] [.reference_item]
[ .reference_list ]
相关推荐












