首页 文章

Keras致密层形状不匹配

提问于
浏览
1

我想在Keras中创建一个多类分类器,但是我在Dense层中得到了一个维度不匹配 .

MAX_SENT_LENGTH = 100
MAX_SENTS = 15
EMBEDDING_DIM = 100

x_train = data[:-nb_validation_samples]
y_train = labels[:-nb_validation_samples]
x_val = data[-nb_validation_samples:]
y_val = labels[-nb_validation_samples:]

embedding_layer = Embedding(len(word_index) + 1,
                            EMBEDDING_DIM,
                            weights=[embedding_matrix],
                            input_length=MAX_SENT_LENGTH,
                            trainable=True)

sentence_input = Input(shape=(MAX_SENT_LENGTH,), dtype='int32')
embedded_sequences = embedding_layer(sentence_input)
l_lstm = Bidirectional(LSTM(100))(embedded_sequences)
sentEncoder = Model(sentence_input, l_lstm)

review_input = Input(shape=(MAX_SENTS,MAX_SENT_LENGTH), dtype='int32')
review_encoder = TimeDistributed(sentEncoder)(review_input)
l_lstm_sent = Bidirectional(LSTM(100))(review_encoder)
preds = Dense(7, activation='softmax')(l_lstm_sent)
model = Model(review_input, preds)

model.compile(loss='sparse_categorical_crossentropy',
              optimizer='rmsprop',
              metrics=['acc'])

model.fit(x_train, y_train, validation_data=(x_val, y_val), 
            epochs=10, batch_size=50)

类标签正确转换为1-hot向量,但在尝试拟合模型时,我得到了这个不匹配错误:

('Shape of data tensor:', (5327, 15, 100))
('Shape of label tensor:', (5327, 7))
_________________________________________________________________
Layer (type)                 Output Shape              Param #   
=================================================================
input_2 (InputLayer)         (None, 15, 100)           0         
_________________________________________________________________
time_distributed_1 (TimeDist (None, 15, 200)           351500    
_________________________________________________________________
bidirectional_2 (Bidirection (None, 200)               240800    
_________________________________________________________________
dense_1 (Dense)              (None, 7)                 1407       
=================================================================
Total params: 592,501
Trainable params: 592,501
Non-trainable params: 0
_________________________________________________________________
None
ValueError: Error when checking target: expected dense_1 to have 
              shape (None, 1) but got array with shape (4262, 7)

这个(无,1)维度来自何处以及如何解决此错误?

1 回答

  • 2

    如果您的标签是单热编码,则应使用 loss='categorical_crossentropy' 而不是 loss='sparse_categorical_crossentropy' . 'sparse_categorical_crossentropy' 采用整数标签,这就是需要 (None,1) 维度的原因 .

相关问题