【发布时间】:2017-07-26 14:09:31
【问题描述】:
我是 keras 的新手,我正在尝试对文档进行卷积。每个文档由一个形状为 [40 * 4000] 的矩阵表示(每个文档有 40 个句子,每个句子有 20 个词,因此词嵌入维度为 200)。
我想对 3 个单词进行卷积,这是我的代码(不完全相同,但您可以在您的机器上运行此代码):
import keras
from keras.layers import Input, Embedding, Dense, merge, Conv2D
from keras.layers.core import Reshape
tokens_input = Input(name='input', shape=(40, 20), dtype='int32')
x = Embedding(25000 + 1, 200, name="embedding")(tokens_input)
x = Reshape((40, -1),
name="reshape")(x)
cur_conv = Conv2D(20, kernel_size=(3*200, 1),
strides=(200, 1), padding='valid', activation='relu')(x)
但我收到此错误消息:输入 0 与层 conv2d_1 不兼容:预期 ndim=4,发现 ndim=3
【问题讨论】: