如何对分类预测的结果进行画散点图,请给出具体代码
时间: 2023-06-09 10:02:34 浏览: 197
以下是一个简单的 Python 代码示例,可以将分类预测结果可视化为散点图:
```python
import matplotlib.pyplot as plt
# 假设有两个分类,分别表示为 0 和 1
category0 = [[1, 2], [3, 4], [5, 6]]
category1 = [[2, 1], [4, 3], [6, 5]]
# 生成散点图
plt.scatter([x[0] for x in category0], [x[1] for x in category0], color='red', label='Category 0')
plt.scatter([x[0] for x in category1], [x[1] for x in category1], color='blue', label='Category 1')
# 添加图例和标签
plt.legend()
plt.xlabel('x')
plt.ylabel('y')
plt.title('Classification Prediction Result')
# 显示图像
plt.show()
```
其中,`category0` 和 `category1` 分别表示两个分类的预测结果,每个样本用一个包含两个元素的列表表示。在 `plt.scatter()` 中,将列表中第一个元素作为 x 坐标,第二个元素作为 y 坐标,以不同的颜色区分两个分类的点。在图像中添加一个图例以显示每个分类的标签,同时添加一个标题和标签以提高可读性。最后使用 `plt.show()` 函数显示图像。
阅读全文