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

把keras升级成2.2时报了如下错误,看上去是keras与TensorFlow不兼容了。。。

 

于是打开/anaconda3/lib/python3.6/site-packages/keras/backend/tensorflow_backend.py的3149行

return tf.nn.softmax(x, axis=axis)

改为

return tf.nn.softmax(x)