【发布时间】:2019-03-29 18:10:43
【问题描述】:
我的输入只是一个 441 行 216 列的矩阵:
216 个特征值
441 次试验
总类标签6
我正在尝试在 CNN 模型上训练我的数据:
model = Sequential()
model.add(Conv1D(128, 5, input_shape=(441, 216)))
model.add(Activation('relu'))
model.add(Conv1D(128, 5,padding='same'))
model.add(Activation('relu'))
model.add(Dropout(0.1))
model.add(MaxPooling1D(pool_size=(8)))
model.add(Conv1D(128, 5,padding='same',))
model.add(Activation('relu'))
model.add(Conv1D(128, 5,padding='same',))
model.add(Activation('relu'))
model.add(Conv1D(128, 5,padding='same',))
model.add(Activation('relu'))
model.add(Dropout(0.2))
model.add(Conv1D(128, 5,padding='same',))
model.add(Activation('relu'))
model.add(Flatten())
model.add(Dense(10))
model.add(Activation('softmax'))
opt = keras.optimizers.rmsprop(lr=0.00001, decay=1e-6)
model.compile(loss='categorical_crossentropy',
optimizer=opt,metrics=['accuracy'])
这会引发错误: ValueError: 检查模型输入时出错:预期 conv1d_1_input 的形状为 (None, 441, 216) 但得到的数组的形状为 (1, 441, 216)
如何将我的输入输入到 CNN 中?
【问题讨论】:
标签: python tensorflow keras