sklearn贝叶斯分类对结果进行可视化
时间: 2023-06-29 16:09:29 浏览: 206
我们可以使用matplotlib库来进行结果可视化。具体步骤如下:
1. 将数据集分为训练集和测试集,并进行特征提取和预处理。
2. 导入贝叶斯分类器,并进行模型训练。
3. 对测试集进行预测,并计算准确率。
4. 使用matplotlib库绘制出测试集中真实标签和预测标签的比较图。
下面是一个示例代码:
```python
from sklearn import datasets
from sklearn.model_selection import train_test_split
from sklearn.naive_bayes import GaussianNB
import matplotlib.pyplot as plt
# 导入数据集
iris = datasets.load_iris()
X = iris.data
y = iris.target
# 将数据集分为训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=0)
# 特征提取和预处理
# ...
# 导入贝叶斯分类器,并进行模型训练
gnb = GaussianNB()
gnb.fit(X_train, y_train)
# 对测试集进行预测,并计算准确率
y_pred = gnb.predict(X_test)
accuracy = gnb.score(X_test, y_test)
# 使用matplotlib库绘制出测试集中真实标签和预测标签的比较图
plt.scatter(X_test[:, 0], X_test[:, 1], c=y_test)
plt.scatter(X_test[:, 0], X_test[:, 1], c=y_pred, marker='x')
plt.title('Naive Bayes Classification (Accuracy: %f)' % accuracy)
plt.xlabel('Feature 1')
plt.ylabel('Feature 2')
plt.show()
```
在上述代码中,我们使用散点图来表示测试集中的数据点。其中,蓝色点表示真实标签,红色叉号表示预测标签。图中的标题显示了准确率。您可以根据需要更改标题、标签和颜色等属性。
阅读全文