3

我想在 tfLearn 中构建循环神经网络二元分类器。但我想使用我保存在泡菜中的预先构建的词嵌入。以下是我在模型中使用的 tflearn 代码。

# Network building
net = tflearn.input_data([None, 100])
net = tflearn.embedding(net, input_dim=10000, output_dim=128) ##Have to change this##
net = tflearn.lstm(net, 128, dropout=0.8)
net = tflearn.fully_connected(net, 2, activation='softmax')
net = tflearn.regression(net, optimizer='adam', learning_rate=0.001,
                         loss='categorical_crossentropy')

如何在此模型中使用我预先构建的词嵌入(第 2 行)?

4

0 回答 0