sklearn实现贝叶斯分类预测数据,并计算每项预测数据的准确率
时间: 2023-08-12 07:13:12 浏览: 121
使用sklearn对多分类的每个类别进行指标评价操作
5星 · 资源好评率100%
首先,我们需要将数据分为训练集和测试集。
假设我们已经将数据分为了训练集和测试集,然后我们使用朴素贝叶斯分类器对测试集进行预测,并计算每项预测数据的准确率。
以下是一个简单的示例代码:
```python
from sklearn.naive_bayes import GaussianNB
from sklearn.metrics import accuracy_score
# 假设我们已经将数据分为了训练集和测试集,分别为X_train, y_train, X_test, y_test
# 创建一个高斯朴素贝叶斯分类器
gnb = GaussianNB()
# 在训练集上拟合模型
gnb.fit(X_train, y_train)
# 对测试集进行预测
y_pred = gnb.predict(X_test)
# 计算预测准确率
accuracy = accuracy_score(y_test, y_pred)
print("预测准确率:", accuracy)
```
在这个示例中,我们使用了`GaussianNB`类来创建高斯朴素贝叶斯分类器,并使用`fit`方法在训练集上拟合模型。然后,我们使用`predict`方法对测试集进行预测,并使用`accuracy_score`函数计算预测准确率。
阅读全文