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))