compute_test_result.py 文件源码

def get_confusion(gts,pres):
    confusion_mat=np.zeros([101,101])
    len_items=len(gts)
    for i in range(len_items):
        confusion_mat[gts[i]][pres[i]]+=1
    for i in range(101):
        confusion_mat[i,:]/=sum(confusion_mat[i,:])
    #embed()
    return confusion_mat