如果您有两个包含预测值和实际值的列表;看起来就像您所做的那样,您可以将它们传递给一个函数,该函数将使用以下内容来计算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)
我认为您可以从这里计算出您感兴趣的利率,以及其他性能指标(例如特异性和敏感性)。
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)