1

我正在学习如何创建顺序模型。我有一个模型：

*model  =  Sequential()*

然后我继续添加池化层和卷积层（这很好）。但是在创建密集层时：

*model.add(Dense(num_classes, activation = 'softmax'))*

返回的行：

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

由于未定义轴而导致错误。Keras 和 TensorFlow 文档都显示 softmax 的默认轴是 None 或 -1。

这是keras的错误吗？是否有一个简单的解决方法（如果我要设置轴，我不确定输入张量是什么）？

- 如果需要，我可以添加其余代码，但它只是由其他层组成，我认为它不会有太大帮助。

4

1 回答 1

2

我相信您的 Keras 和/或 TensorFlow 不是最新的，您应该更新它/它们。

这是 2017 年夏季 Keras 中的一个已知问题，并已在此提交中修复。请参阅有关错误报告的此评论的更多信息。

也axis于2017 年 11 月 22 日在 TensorFlow 中作为参数引入，softmax()因此如果 TensorFlow 版本为 1.4.0 或更低，也会导致此错误。

如果您在链接的提交处查看 Keras 的来源，则究竟是哪一个导致错误取决于处理的张量的等级。

此代码适用于当前版本（在https://colab.research.google.com上测试）：

import keras
from keras.models import Sequential
from keras.layers import Dense, Activation
from keras.optimizers import SGD

print( keras.__version__ )

model = Sequential()
model.add( Dense(6, input_shape=(6,), activation = 'softmax' ) )

sgd = SGD(lr=0.01, decay=1e-6, momentum=0.9, nesterov=True)
model.compile(loss='categorical_crossentropy',
              optimizer=sgd,
              metrics=['accuracy'])

输出

2.1.6

但更重要的是，编译模型没有错误。

于 2018-05-14T00:38:11.723 回答