analysis.py

import sys
import path_config
sys.path.append(path_config.get_pese_path())
from ANSAFF.py.tool import signal_detection

def analysis4file(file):
    labs = []
    preds = []
    fh = open(file, 'r')
    i = 0
    for line in fh:
        xline = line[1:len(line)-1]
        xline = xline.split(',')
        xline = [float(i) for i in xline]
        if i % 2 == 0:
            labs.append(xline[0])
        else:
            preds.append(xline[0])
        i += 1
    fh.close()
    tp,fp,tn,fn = signal_detection(labs, preds)
    precision = tp/(tp+fp)
    recall = tp/(tp+fn)
    f1 = 2*precision*recall/(precision+recall)
    print("precision: " + str(precision) + ", recall: " + str(recall) + ", f1: " + str(f1))