如果您有两个包含预测值和实际值的列表;看起来就像您所做的那样,您可以将它们传递给一个函数,该函数将使用以下内容来计算TP,FP,TN,FN:
def perf_measure(y_actual, y_hat):
TP = 0
FP = 0
TN = 0
FN = 0
for i in range(len(y_hat)):
if y_actual[i]==y_hat[i]==1:
TP += 1
if y_hat[i]==1 and y_actual[i]!=y_hat[i]:
FP += 1
if y_actual[i]==y_hat[i]==0:
TN += 1
if y_hat[i]==0 and y_actual[i]!=y_hat[i]:
FN += 1
return(TP, FP, TN, FN)
我认为您可以从这里计算出您感兴趣的利率,以及其他性能指标(例如特异性和敏感性)。