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)