最近在计算AUC时,最后的图片无法输出,提示“No handles with labels found to put in legend.”,检查发现是没有定义图例,于是增加图例后解决。
37 plt.figure()
38 plt.plot(recall, precision, label='AUC')
39 plt.xlim([0.0, 1.0])
40 plt.ylim([0.0, 1.05])
41 plt.xlabel('recall')
42 plt.ylabel('precision')
43 plt.title('Precision-Recall curve')
44 plt.legend(loc="lower right")
45 plt.show()
46 plt.savefig('ROC.PNG')
但图片仍然无法直接生成,只能先保存在本地再打开查看,增加plt.savefig('ROC.PNG')实现。后续再研究研究。