keras版本问题TypeError: softmax() got an unexpected keyword argument 'axis'解决方案

TypeError: softmax() got an unexpected keyword argument ‘axis’

  • 解决方案一:回退keras版本,pip install keras==2.1

  • 解决方案二:修改函数softmax()的返回语句
    1.(可选)升级keras:pip install --upgrade keras
    2.(要修改的文件)定位softmax()所在py文件(tensorflow_backend.py)的位置:
    ..\tensorflow\Lib\site-packages\keras\backend\tensorflow_backend.py
    3.(修改内容)编辑其中softmax()函数,修改return tf.nn.softmax(x, axis=axis)为:

return tf.nn.softmax(x, axis)
发布了22 篇原创文章 · 获赞 21 · 访问量 7019

猜你喜欢

转载自blog.csdn.net/weixin_43108122/article/details/100040988