【发布时间】:2017-12-19 22:46:27
【问题描述】:
一直在尝试在 Keras 中创建神经网络,但遇到了一个问题,即我的一个密集层和激活层之间的形状不匹配。我错过了一些明显的东西吗?使用 TensorFlow 后端。
print(x_train.shape)
print(y_train.shape)
(1509, 476, 4)
(1509,)
那么我的模型如下:
###Setup Keras to create a bidirectional convolutional recurrent NN based on DanQ NN
###See https://github.com/uci-cbcl/DanQ
model = Sequential()
model.add(Conv1D(filters=320,
kernel_size=26,
padding="valid",
activation="relu",
strides=1,
input_shape=(476, 4)
))
model.add(MaxPooling1D(pool_size=13, strides=13))
model.add(Dropout(0.2))
model.add(keras.layers.wrappers.Bidirectional(LSTM(320, return_sequences=True, input_shape=(None, 320))))
model.add(Flatten())
model.add(Dense(input_dim=34*640, units=925))
model.add(Activation('relu'))
model.add(Dense(input_dim=925, units=919))
model.add(Activation('sigmoid'))
print('compiling model')
model.compile(loss='binary_crossentropy', optimizer='rmsprop', class_mode="binary")
print('running at most 60 epochs')
model.fit(x_train, y_train.T, batch_size=100, epochs=60, shuffle=True, verbose=2, validation_split=0.1)
tresults = model.evaluate(x_test, y_test, verbose=2)
print(tresults)
print(model.output_shape)
但我收到以下错误:
ValueError: Error when checking target: expected activation_48 to have shape (None, 919) but got array with shape (1509, 1)
错误似乎源于使用 sigmoid 激活输入到第二个激活层。例如:
model.add(Dense(input_dim=925, units=919))
model.add(Activation('sigmoid'))
为什么会出现不匹配?
【问题讨论】:
-
您不必在密集层中指定 input_dim 和单位。前任。最后一个密集层应该是
model.add(Dense(1,activation='sigmoid')) -
谢谢!这似乎解决了不匹配问题,但弹出了一个新错误:
ValueError: Invalid argument "class_mode" passed to K.function with Tensorflow backend,源自model.fit()。对此有什么想法吗? -
是的,你不需要在
model.compile()中有参数class_mode='binary'
标签: python tensorflow neural-network keras reshape