如何在tfLearn中使用预建的单词嵌入?

时间:2017-02-05 15:23:54

标签: tensorflow recurrent-neural-network tflearn

我想在tfLearn中构建递归神经网络二进制分类器。但我想使用我预先构建的单词嵌入,这是我在pickle中保存的。以下是我在模型中使用的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行)?

0 个答案:

没有答案