Performance metric

performance_metric.py

import numpy as np
def pf(output,target,metric=None):
    TP = np.count_nonzero(data*target)
    TN = np.count_nonzero((data - 1) * (target - 1))
    FP = np.count_nonzero(data * (target - 1))
    FN = np.count_nonzero((data - 1) * target)
    precision = TP / (TP + FP)
    recall = TP / (TP + FN)
    F1 = 2 * precision * recall / (precision + recall)
    accuracy = (TP+TN)/(TP+TN+FP+FN)
    PPV = TP/(TP+FP)
    if metric=='precision':
        return precision
    elif metric=='recall':
        return recall
    elif metric=='PPV':
        return PPV
    elif metric=='accuracy':
        return accuracy
    else: return F1