這次使用的是交叉熵?fù)p失函數(shù),該函數(shù)在分類任務(wù)上比較常用。
定義了一個損失函數(shù)之后,還有對它求平均值,訓(xùn)練程序必須返回平均損失作為第一個返回值,因為它會被后面反向傳播算法所用到。。
同時我們還可以定義一個準(zhǔn)確率函數(shù),這個可以在我們訓(xùn)練的時候輸出分類的準(zhǔn)確率。
In[14]