【发布时间】:2019-03-07 20:52:42
【问题描述】:
如何在语料库上运行 fasttext 并在 mxnet 符号嵌入层中使用这些嵌入?
【问题讨论】:
标签: mxnet
如何在语料库上运行 fasttext 并在 mxnet 符号嵌入层中使用这些嵌入?
【问题讨论】:
标签: mxnet
为此,您首先需要加载包含 FastText 嵌入的矩阵,然后将其作为初始化程序传递给嵌入层:
embed_layer_3 = mx.sym.Embedding(data=input_x_3, weight=the_emb_3, input_dim=vocab_size, output_dim=embedding_dim, name='vocab_embed')
这个例子来自here,他们使用 Glove Embedding,但想法是一样的。
我强烈建议使用 Gluon API 而不是 Symbol API。在这种情况下,您可以更轻松地使用 GluonNLP 包的所有优点,该包已经预训练了 FastText 嵌入。请参阅this tutorial 了解如何在 GluonNLP 中使用 Fasttext
【讨论】: