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