1、绘制直线图
# 训练过程可视化
plt.figure("Training Cost")
plt.title("Training Cost", fontsize=24)
plt.xlabel("iter", fontsize=14)
plt.ylabel("cost", fontsize=14)
plt.plot(iters, train_costs, color="red", label="Training Cost")
plt.grid()
plt.savefig("train.png")
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/d11039efa5dd592fc4f76eaa19a6cadb.png)
2、绘制散点图
# 可视化
plt.figure('scatter')
plt.title("TestFigure", fontsize=24)
plt.xlabel("ground truth", fontsize=14)
plt.ylabel("infer result", fontsize=14)
x = np.arange(1, 30)
y = x
plt.plot(x, y)
plt.scatter(ground_truths, infer_result, color="green", label="Test")
plt.grid()
plt.legend()
plt.savefig("predict.png")
plt.show()
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/4ebc5c1dc22a5c7e8f60e88d423cbf62.png)