pytorch中訓練完網絡后，需要對學習的結果進行測試。官網上例程用的方法統(tǒng)統(tǒng)都是正確率，使用的是torch.eq()這個函數。

從網站建設到定制行業(yè)解決方案，為提供網站制作、成都網站設計服務體系，各種行業(yè)企業(yè)客戶提供網站建設解決方案，助力業(yè)務快速發(fā)展。成都創(chuàng)新互聯將不斷加快創(chuàng)新步伐，提供優(yōu)質的建站服務。

但是為了更精細的評價結果，我們還需要計算其他各個指標。在把官網API翻了一遍之后發(fā)現并沒有用于計算TP，TN，FP，FN的函數。。。

在動了無數歪腦筋之后，心想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

另外有需要云服務器可以了解下創(chuàng)新互聯scvps.cn，海內外云服務器15元起步，三天無理由+7*72小時售后在線，公司持有idc許可證，提供“云服務器、裸金屬服務器、高防服務器、香港服務器、美國服務器、虛擬主機、免備案服務器”等云主機租用服務以及企業(yè)上云的綜合解決方案，具有“安全穩(wěn)定、簡單易用、服務可用性高、性價比高”等特點與優(yōu)勢，專為企業(yè)上云打造定制，能夠滿足用戶豐富、多元化的應用場景需求。

本文標題：在pytorch中計算精度、回歸率、F1score等指標的實例-創(chuàng)新互聯
網頁地址：http://www.chinadenli.net/article8/gchip.html

成都網站建設公司_創(chuàng)新互聯，為您提供域名注冊、做網站、外貿建站、企業(yè)網站制作、網站營銷、App設計

廣告

聲明：本網站發(fā)布的內容（圖片、視頻和文字）以用戶投稿、用戶轉載內容為主，如果涉及侵權請盡快告知，我們將會在第一時間刪除。文章觀點不代表本網站立場，如需處理請聯系客服。電話：028-86922220；郵箱：631063699@qq.com。內容未經允許不得轉載，或轉載時需注明來源： 創(chuàng)新互聯