prcurve.py

def prcurve(y,pred):
    P = []
    R = []
    thresholds = np.arange(0.0,1.0+0.01,0.01)
    for th in thresholds:
        P.append(precision(y,pred,th))
        R.append(recall(y,pred,th))
    plt.figure(figsize=(10,10))
    plt.plot(R,P,linewidth=2)
    plt.xlim(0.0, 1.0)
    plt.ylim(0.0, 1.0)
    plt.xlabel("Recall",fontsize=12)
    plt.ylabel("Precision",fontsize=12)
    plt.legend(fontsize=10, loc='best')
    plt.title("Precision Recall Curve",fontsize=12)
    plt.savefig('Precision_Recall_Curve.png')