原文:對於計算正確率時 logits.argmax(dim=1),torch.eq(pre_label,label)

額 好像是一句非常簡單的代碼 ,但是作為新手 ,我是完全看不懂哎 前十眼。 首先 這里的logits是一個 a,b 維的張量。其中a是你的全連接輸出維度,b是一個batch中的樣本數量。 我們經過一個argmax的操作,dim 意味着找到張量中各自的最大值所在索引。也就是找到每個樣本的全連接輸出中最大的那一個。 最有可能的預測值。 torch.eq會返回一個 batch維的bool值。 。sum統 ...

2021-06-04 17:06 0 1044 推薦指數:

查看詳情

torch.argmaxdim詳解

torch.argmax()函數 argmax函數:torch.argmax(input, dim=None, keepdim=False)返回指定維度最大值的序號,dim給定的定義是:the demention to reduce.也就是把dim這個維度的,變成這個維度的最大值的index ...

Thu Jul 22 02:17:00 CST 2021 0 516
機器學習 | 分類性能度量指標 : ROC曲線、AUC值、正確率、召回

本篇博客的圖源來自 zhwhong,轉載僅作學習使用! 在分類任務中,人們總是喜歡基於錯誤來衡量分類器任務的成功程度。錯誤指的是在所有測試樣例中錯分的樣例比例。實際上,這樣的度量錯誤掩蓋了樣例如何被分錯的事實。在機器學習中,有一個普遍適用的稱為混淆矩陣(confusion ...

Tue Oct 26 06:29:00 CST 2021 0 305
新手學信息檢索5:正確率-召回與搜索引擎的評價

這篇就簡單介紹一下搜索引擎的評價方法。從用戶的角度去評價一個搜索引擎的檢索效果最好的方法就是計算用戶在查到自己滿意文檔已經瀏覽的文檔數。但是實際中,查詢千變萬化,文檔也千變萬化,所以這種方法不可行。人們便提出了下面的概念,並建立了一個評價標准。 這里面有三個常用的概念:正確率,精確,召回 ...

Wed Jan 09 19:17:00 CST 2013 0 3131
 
粵ICP備18138465號   © 2018-2026 CODEPRJ.COM