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

 

相关文章:

  • 2022-01-07
  • 2021-10-08
  • 2021-06-16
  • 2022-02-12
  • 2022-12-23
  • 2021-07-28
  • 2021-11-19
猜你喜欢
  • 2021-10-29
  • 2021-06-05
  • 2022-12-23
  • 2022-12-23
  • 2021-09-27
  • 2021-12-10
  • 2021-06-21
相关资源
相似解决方案