我正在尝试使用 keras 制作预测模型答案

【问题标题】：I'm trying to make a prediction model using keras我正在尝试使用 keras 制作预测模型
【发布时间】：2019-07-12 08:55:11
【问题描述】：

当我尝试在 keras 中创建卷积神经网络并且想要检查我的模型时，我收到了这个错误：

init() 缺少 1 个必需的位置参数：'units'

我使用的keras版本是2.2.4版

这是模型：

input_shape = (224, 224, 1)


model = Sequential()
model.add(Conv2D(16, kernel_size=(5, 5), strides=2, padding='same', activation='relu', input_shape=input_shape))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Conv2D(32, kernel_size=(5, 5), strides=1, padding='same', activation='relu'))
model.add(MaxPooling2D(pool_size=(4, 4)))
model.add(Conv2D(64, kernel_size=(5, 5), strides=1, padding='same', activation='relu'))
model.add(MaxPooling2D(pool_size=(4, 4)))
model.add(Flatten())
model.add(Dense(1024, activation='relu'))
model.add(Dense(activation='linear'))

model.compile(loss='mse',
              optimizer='adam')


model.summary()

错误是：

init() 缺少 1 个必需的位置参数：'units'

【问题讨论】：

标签： python-3.x keras conv-neural-network

【解决方案1】：

问题出在你的最后一行：

model.add(Dense(activation='linear'))

你需要为你的密集层指定一个大小，所以你可以这样做：

model.add(Dense(1024, activation='relu'))
model.add(Dense(NEWSIZE, activation='linear')) #

NEWSIZE 是您想要的单位数

或者这个：

model.add(Dense(1024, activation='relu'))
model.add(Activation('linear'))

【讨论】：