python - [Python][Keras] softmax() 获得意外的关键字参数 'axis'

Question

我正在尝试运行以下代码来创建卷积神经网络。该代码来 self 找到的教程。

from keras.models import Sequential
from keras.layers import Dense, Conv2D, Flatten
model = Sequential()
model.add(Conv2D(64, kernel_size=3, activation='relu', input_shape=(28,28,1)))
model.add(Conv2D(32, kernel_size=3, activation='relu'))
model.add(Flatten())
model.add(Dense(10, activation='softmax'))

我已经寻找过答案，但没有任何效果。有人说降级 Keras，有人说用“pip install -U keras tensorflow”升级它，但我仍然收到相同的错误消息。

有人知道如何解决这个问题吗？非常感谢!

Answer 1

肯定是你使用的tensorflow和Keras的版本有问题。我的假设是您首先安装了tensorflow，然后安装了keras。 Softmax 相对较新(可能是今年早些时候)。您的tensorflow版本可能是添加Softmax之前的版本，但Keras的版本是添加之后的版本。因此，当您调用 Softmax 时，keras 会尝试在 Tensorflow 中调用它，但会收到错误。

你需要做什么
您应该将 Keras 和 Tensorflow 都更新到当前版本，并仔细检查版本。如果您使用的是 Anaconda 等包管理器，即使在更新 Keras 和 Tensorflow 之后，您也可能使用旧版本。因此，请仔细检查您是否使用的是当前版本