precision recall and auc plot

auc_and_precision_recall_plot.py

def plot_auc_pr(y_test, y_pred):
    
    fig = plt.figure(figsize=(20,7))

    plt.subplot(1,2,1)
    fpr, tpr, _ = roc_curve(y_test, y_pred)
    auc = roc_auc_score(y_test, y_pred)
    plt.plot(fpr,tpr,label="auc="+str(auc))
    plt.title('roc-auc curve')
    plt.legend(loc=4)
    
    plt.subplot(1,2,2)
    pr, rc, _ = precision_recall_curve(y_test, y_pred)
    auc = roc_auc_score(y_test, y_pred)
    plt.plot(pr,rc,label="auc="+str(auc))
    plt.title('precision recall curve')
    plt.xlabel("precision")
    plt.ylabel("recall")
    plt.legend(loc=1 )

    plt.show()