在pytorch 中計算精度、回歸率、F1 score等指標的實例
更新時間：2020年01月18日 11:20:52 作者：Link2Link
今天小編就為大家分享一篇在pytorch 中計算精度、回歸率、F1 score等指標的實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
pytorch中訓(xùn)練完網(wǎng)絡(luò)后，需要對學(xué)習(xí)的結(jié)果進行測試。官網(wǎng)上例程用的方法統(tǒng)統(tǒng)都是正確率，使用的是torch.eq()這個函數(shù)。
但是為了更精細的評價結(jié)果，我們還需要計算其他各個指標。在把官網(wǎng)API翻了一遍之后發(fā)現(xiàn)并沒有用于計算TP，TN，F(xiàn)P，F(xiàn)N的函數(shù)。。。
在動了無數(shù)歪腦筋之后，心想pytorch完全支持numpy，那能不能直接進行判斷，試了一下果然可以，上代碼：
# TP predict 和 label 同時為1 TP += ((pred_choice == 1) & (target.data == 1)).cpu().sum() # TN predict 和 label 同時為0 TN += ((pred_choice == 0) & (target.data == 0)).cpu().sum() # FN predict 0 label 1 FN += ((pred_choice == 0) & (target.data == 1)).cpu().sum() # FP predict 1 label 0 FP += ((pred_choice == 1) & (target.data == 0)).cpu().sum() p = TP / (TP + FP) r = TP / (TP + FN) F1 = 2 * r * p / (r + p) acc = (TP + TN) / (TP + TN + FP + FN
這樣就能看到各個指標了。
因為target是Variable所以需要用target.data取到對應(yīng)的tensor，又因為是在gpu上算的，需要用 .cpu() 移到cpu上。
因為這是一個batch的統(tǒng)計，所以需要用+=累計出整個epoch的統(tǒng)計。當然，在epoch開始之前需要清零
以上這篇在pytorch 中計算精度、回歸率、F1 score等指標的實例就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
torch.nn是包含了構(gòu)筑神經(jīng)網(wǎng)絡(luò)結(jié)構(gòu)基本元素的包,在這個包中可以找到任意的神經(jīng)網(wǎng)絡(luò)層,下面這篇文章主要給大家介紹了關(guān)于PyTorch中torch.nn.Linear的相關(guān)資料,文中通過實例代碼介紹的非常詳細,需要的朋友可以參考下2022-06-06
這篇文章主要為大家介紹了python實現(xiàn)多線程及線程間通信的簡單方法示例，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪2023-07-07
這篇文章主要介紹了Python字符串三種格式化輸出,需要的朋友可以參考下2020-09-09
這篇文章主要介紹了python3.9之你應(yīng)該知道的新特性詳解,需要的朋友可以參考下2021-04-04
Python的threading模塊中提供了多種鎖的相關(guān)方法,Python的多線程不能同時執(zhí)行,因而鎖的使用非常關(guān)鍵,下面我們就來舉例講解Python編程中對線程鎖的使用:2016-07-07
這篇文章主要介紹了Django實現(xiàn)圖片上傳功能步驟解析,文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2020-04-04
這篇文章主要介紹了詳解django自定義中間件處理，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2018-11-11
在本文里我們給大家整理了關(guān)于通過API管理或定制開發(fā)ECS的相關(guān)實例內(nèi)容，有需要的朋友們參考學(xué)習(xí)下。2018-09-09
這篇文章主要介紹了python?import模塊時有錯誤紅線的原因及解決，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-02-02
這篇文章主要介紹了Python設(shè)計模式編程之橋接模式的運用,橋接模式主張把抽象部分與它的實現(xiàn)部分分離,需要的朋友可以參考下2016-03-03