利用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