Question

我正在尝试运行以下代码以创建卷积神经网络。该代码来自我发现的一个教程。

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之后，您可能仍在使用旧版本。因此，请仔细检查您使用的是当前版本

[Python] [Keras] softmax（）得到了意外的关键字参数'axis'

1 个答案: