深度學習 基於pytorch框架的分類問題損失函數的選擇NLL

前提:輸出層的激活函數爲softmax(存在問題)      分類問題可以使用MSE作爲損失函數,但Pytorch官方提供了一種NLL(negative log likelihood)損失函數:                                                                          在上式中out爲輸出層的輸出,假設爲[[0.3, 0.7]
相關文章
相關標籤/搜索