ROC曲线绘制注意事项和小技巧

简介

假设你已经把所有曲线绘制在了plt图像上。
参考这篇文章：
ROC曲线绘制和求分类四个指标

代码

plt.plot([0, 1], [0, 1], 'k--')  # 绘制对角线
plt.xlabel('False Positive Rate')
plt.ylabel('True Positive Rate')
plt.legend(loc='lower right')
plt.gca().set_aspect('equal',adjustable='box')
plt.xlim([-0.020,1.000])
plt.ylim([-0.002,1.02])
plt.savefig('myROC.jpg',dpi=600)
plt.show()

XY轴范围设置

plt.xlim([-0.020,1.000])
plt.ylim([-0.002,1.02])

设置横纵轴一样长

plt.gca().set_aspect('equal',adjustable='box')

可以解决ROC图像扁平的情况。

分辨率和保存

plt.savefig('myROC.jpg',dpi=600)

这样图片会清晰很多。