TypeError:softmax() got an unexpected keyword argument 'axis'


利用MNIST數據集做手寫數字識別時,model.add(Dense(units=10,activation='softmax')) 一直報如下錯:

TypeError:softmax() got an unexpected keyword argument 'axis'。

查了下相關主題。有以下幾種解決方案:

1)keras 2.1.6 版本中softmax已經沒有axis參數,將keras版本降為2.1后問題解決;但問題是在降級版本安裝后提示會與keras的其他模塊不匹配,可能會帶來其它的問題,因此這不是一個好的解決方案。

2)刪除tensorflow_backend.py文件 line 2963中

return tf.nn.softmax(x,axis=axis)

中的axis=axis去掉。試過了不起作用。

3)查stackflow中相關主題,了解到這一信息:tensorflow1.4或更低版本會產生此問題。說明下,本人裝的是Anaconda,查了下所裝tensorflow版本,為1.1。果斷升級至1.8.0。再嘗試相關代碼,搞定。語句如下:

pip install tensorflow==1.8.0

參考:https://blog.csdn.net/bailianfa/article/details/80891051

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM