使用Python实现朴素贝叶斯分类器
时间: 2024-03-06 11:45:17 浏览: 211
python实现朴素贝叶斯分类器
5星 · 资源好评率100%
朴素贝叶斯分类器是一种基于贝叶斯定理和特征条件独立性假设的分类算法。在Python中,可以使用scikit-learn库来实现朴素贝叶斯分类器。
以下是使用Python实现朴素贝叶斯分类器的步骤:
1. 导入所需的库:
```python
from sklearn.naive_bayes import GaussianNB
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score
```
2. 准备数据集:
准备一个包含特征和对应标签的数据集,通常将数据集分为训练集和测试集。
3. 创建朴素贝叶斯分类器对象:
```python
model = GaussianNB()
```
4. 拟合模型:
使用训练集对模型进行拟合,即学习训练数据的概率分布。
```python
model.fit(X_train, y_train)
```
5. 进行预测:
使用训练好的模型对测试集进行预测。
```python
y_pred = model.predict(X_test)
```
6. 评估模型:
使用评估指标(如准确率)来评估模型的性能。
```python
accuracy = accuracy_score(y_test, y_pred)
```
以上就是使用Python实现朴素贝叶斯分类器的基本步骤。
阅读全文