在二分类问题中,准确率一直处于50%上下的解决方法
生活随笔
收集整理的這篇文章主要介紹了
在二分类问题中,准确率一直处于50%上下的解决方法
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
最近在 貓狗識別 項目中,不知為何準確率(訓練集準確率和驗證集準確率)一直處于0.5左右,這說明網絡根本沒有學習。后來查閱了許多他人的經驗,并做了總結。
首先談談我的是如何解決的:
- 網絡結構:AlexNet
- 優化器:Adam (這里正是問題所在)
- 框架:PyTorch
關于優化器,我最開始使用的的是 torch.optim.Adam(),但是訓練100個Epoch后,準確率始終在 49% ~ 50%。
解決方法: 換個優化器!我使用的是 torch.optim.SGD(),其他的或許也行,但我還沒有嘗試。
Note:
出現以上問題時,我們會發現 loss 是 0.69,這是因為輸出 0, 1 的概率都是0.5,而 \(-\ln 0.5=0.69\) 。
網絡上也有一些解決方法,可以對照檢查自己的代碼:
轉載于:https://www.cnblogs.com/xxxxxxxxx/p/11508020.html
總結
以上是生活随笔為你收集整理的在二分类问题中,准确率一直处于50%上下的解决方法的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: Protobuf 安装及 Python、
- 下一篇: 强化学习——入门