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

Try this:

import tensorflow as tf 

Then add a softmax layer in this way:

model.add(Activation(tf.nn.softmax))

Leave a Comment