python画pr曲线_根据Precision和recall结果画PR曲线图

写在前面：这是自己的学习笔记，如有侵权，请联系。

先上代码！

# coding:utf-8

import cPickle

import os

import matplotlib as mpl

import matplotlib.pyplot as plt

mpl.rcParams[u'font.sans-serif'] = ['simhei']

mpl.rcParams['axes.unicode_minus'] = False #用来正常显示负号

recall_pkl = "/media/wj/wj/yolo_learn/dateset_deal/recall.pkl"

precison_pkl = "/media/wj/wj/yolo_learn/dateset_deal/precision.pkl"

recall = cPickle.load(open(recall_pkl))

precision = cPickle.load(open(precison_pkl))

clr = ['r', 'g', 'b', 'k']

classitem = [u'瓦楞纸盒',u'文件袋', u'塑料包裹',u'编织袋']

model = ['Real','Real+Synthetic Adapted']

for j in range(1, len(recall)/2+1):

plt.figure()

plt.plot(recall[j - 1], precision[j - 1], color=clr