【发布时间】:2021-10-14 14:02:52
【问题描述】:
我正在使用 LSTM 进行假新闻检测,并在我的模型中添加了嵌入层。
没有在 LSTM 函数中添加任何 input_shape 就可以正常工作,但我认为input_shape 参数是强制性的。有人可以帮我解释为什么即使没有定义input_shape 也没有错误?是不是因为嵌入层隐式定义了input_shape?
以下是代码:
model=Sequential()
embedding_layer = Embedding(total_words, embedding_dim, weights=[embedding_matrix], input_length=max_length)
model.add(embedding_layer)
model.add(LSTM(64,))
model.add(Dense(1,activation='sigmoid'))
opt = SGD(learning_rate=0.01,decay=1e-6)
model.compile(loss = "binary_crossentropy", optimizer = opt,metrics=['accuracy'])
model.fit(data,train['label'], epochs=30, verbose=1)
【问题讨论】:
标签: python tensorflow keras deep-learning lstm