ROC CURVE

roc.py

def roc(pred, actual):
    roc=[]
    p=float(len(actual)-sum(actual))
    n=float(sum(actual))
    for i in range(0,100,1) :
        TP=0
        FP=0
        i = float(i)/100
        for j in range(len(pred)):
            if (pred[j] > i) & (actual[j]==0):
                TP+=1
            if (pred[j] > i) & (actual[j]==1):
                FP+=1
        roc.append((TP/p,FP/n))
    return roc