無論是官方文檔還是各位大神的論文或搭建的網(wǎng)絡(luò)很多都是計(jì)算準(zhǔn)確率，很少有計(jì)算誤判率，

下面就說說怎么計(jì)算準(zhǔn)確率以及誤判率、召回率等指標(biāo)

1.計(jì)算正確率

獲取每批次的預(yù)判正確個(gè)數(shù)

train_correct = (pred == batch_y.squeeze(1)).sum()

該語句的意思是 預(yù)測的標(biāo)簽與實(shí)際標(biāo)簽相等的總數(shù)

獲取訓(xùn)練集總的預(yù)判正確個(gè)數(shù)

train_acc += train_correct.data[0] #用來計(jì)算正確率

準(zhǔn)確率 ： train_acc / (len(train_data))

2.誤判率

舉例：當(dāng)你是二分類時(shí)，你需要計(jì)算 原標(biāo)簽為1，但預(yù)測為 0 ，以及 原標(biāo)簽為0，預(yù)測為1的 誤判率

誤判率又分為：

CTW : correct to wrong 標(biāo)簽為正確的，預(yù)測為錯(cuò)誤的

WTC: wrong to correct 標(biāo)簽為錯(cuò)誤的，預(yù)測為正確的

zes=Variable(torch.zeros(lasize).type(torch.LongTensor))#全0變量

ons=Variable(torch.ones(lasize).type(torch.LongTensor))#全1變量

train_correct01 = ((pred==zes)&(batch_y.squeeze(1)==ons)).sum() #原標(biāo)簽為1，預(yù)測為 0 的總數(shù)

train_correct10 = ((pred==ons)&(batch_y.squeeze(1)==zes)).sum() #原標(biāo)簽為0，預(yù)測為1 的總數(shù)

train_correct11 = ((pred_y==ons)&(batch_y.squeeze(1)==ons)).sum()
train_correct00 = ((pred_y==zes)&(batch_y.squeeze(1)==zes)).sum()

獲取訓(xùn)練集總的誤判個(gè)數(shù)

FN += train_correct01.data[0]

FP += train_correct10.data[0]

TP += train_correct11.data[0]
TN += train_correct00.data[0]

誤判率 ：

(FN+FP)/(len(train_data)) #CTW+WTC

3.精準(zhǔn)率和召回率

精準(zhǔn)率： P = TP/ (TP+FP)
召回率： R = TP/ (TP+FN)

4.真正例率和假正例率

真正例率：TPR = TP/ (TP+FN)
假正例率：FPR =FP/ (FP+TN)

最后，當(dāng)你要計(jì)算多分類的誤判率時(shí)，只需在二分類的基礎(chǔ)上類推即可

以上這篇Pytorch 計(jì)算誤判率,計(jì)算準(zhǔn)確率,計(jì)算召回率的例子就是小編分享給大家的全部內(nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

最新評論