LSTM架构将文本分类在30个类别内,问题是在3个时代之后是否过度拟合,有没有避免过度拟合的建议
model.add(Embedding(max_words, embedding_dimension, input_length=X.shape[1]))
model.add(LSTM(100, return_sequences=True))
model.add(LSTM(100))
model.add(Dense(30, activation='sigmoid'))
model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['acc'])
print(model.summary())
目前没有回答
相关问题 更多 >
编程相关推荐