【发布时间】:2020-10-10 12:52:18
【问题描述】:
我想将 CNN 层的输出馈送到 LSTM 层,但出现错误ValueError: Input 0 is incompatible with layer lstm_15: expected ndim=3, found ndim=2,代码如下:
inp = Input(shape = (max_length,))
xe = Embedding(vocabulary_size, 300, weights = [embedding_matrix], trainable = False)(inp)
x = Conv1D(512,kernel_size = 2, activation='relu',kernel_initializer = "he_uniform")(xe)
x = GlobalMaxPooling1D()(x)
x = LSTM(128)(x)
x = Dense(11, activation = "sigmoid")(x)
输入形状:
embedding_matrix: (26441, 300)
inp : TensorShape([Dimension(None), Dimension(3146)])
X_train :(1432, 3146)
Y_train: (1432, 11)
vocabulary_size: 26441
max_length: 3146
有人可以帮助我
【问题讨论】:
标签: python keras lstm text-classification conv-neural-network