【发布时间】:2021-02-02 21:35:00
【问题描述】:
我花了一些时间来提高我的多类文本分类任务的 F1 分数。我从笔记本电脑评论中提取方面和情绪。因此有 3 个标签,B_A / I_A / O 等。我非常感谢任何改进我的网络的建议,例如附加层或另一个嵌入。 (也许我还应该为我的任务尝试多类分类以外的其他方法) 现在,我的以下代码的 F1 分数约为 60%:
#vocab_size=4840, embedding is glove6B, max_seq_length=100
model = Sequential()
model.add(Embedding(vocab_size, 300, weights=[embedding_vectors], input_length=max_seq_length,
trainable= False))
model.add(Dropout(0.1))
model.add(Conv1D(3000, 1, activation='relu'))
model.add(Bidirectional(LSTM(units=150, recurrent_dropout=0, return_sequences=True)))
model.add(Dense(32, activation='relu'))
model.add(Dense(n_tags, activation='softmax'))
model.compile(loss="categorical_crossentropy", optimizer="rmsprop", metrics=["categorical_accuracy"])
model.summary()
# fit model on train data
model.fit(x_train, y_train,
batch_size=64,
epochs=10)
【问题讨论】:
标签: tensorflow keras lstm stanford-nlp layer