p ython朴素贝叶斯分类
时间: 2023-07-11 17:14:50 浏览: 105
好的,Python中的朴素贝叶斯分类器可以通过使用scikit-learn库来实现。下面是一个简单的代码示例:
首先,安装scikit-learn库:
```
pip install scikit-learn
```
然后,导入必要的库并加载数据集:
```python
from sklearn.datasets import load_iris
from sklearn.naive_bayes import GaussianNB
# 加载数据集
iris = load_iris()
X = iris.data
y = iris.target
```
接下来,创建一个朴素贝叶斯分类器并拟合数据:
```python
# 创建分类器
gnb = GaussianNB()
# 拟合数据
gnb.fit(X, y)
```
最后,使用分类器进行预测:
```python
# 预测新数据
new_data = [[5.1, 3.5, 1.4, 0.2], [6.2, 3.4, 5.4, 2.3]]
print(gnb.predict(new_data))
```
上面的代码将使用朴素贝叶斯分类器对两个新数据进行分类预测,并输出预测结果。
相关问题
p ython朴素贝叶斯算法
Python中的朴素贝叶斯算法是一种基于概率统计的分类模型,它假设特征之间相互独立(即“朴素”),尽管这个假设在实际应用中往往并不成立。朴素贝叶斯算法简单易用,常用于文本分类、垃圾邮件过滤等场景。
在Python中,常用的库如`scikit-learn`提供了一个名为`GaussianNB`的类,用于实现高斯朴素贝叶斯(适用于连续数据),以及`MultinomialNB`和`BernoulliNB`,分别用于处理词频型数据(适合于文档分类)。使用这些库,你可以按照以下步骤进行朴素贝叶斯分类:
1. 导入所需库:
```python
from sklearn.naive_bayes import GaussianNB, MultinomialNB, BernoulliNB
```
2. 准备训练集和测试集:
```python
X_train, y_train = ... # 训练样本特征和标签
X_test, y_test = ... # 测试样本特征和标签
```
3. 初始化并训练模型:
```python
gnb = GaussianNB() # 或者GaussianNB(), MultinomialNB(), BernoulliNB()
gnb.fit(X_train, y_train)
```
4. 预测新数据:
```python
y_pred = gnb.predict(X_test)
```
5. 评估性能:
```python
from sklearn.metrics import accuracy_score
accuracy = accuracy_score(y_test, y_pred)
```
阅读全文