如何在tensorflow.js中构建和训练序列2序列模型

时间:2018-10-14 05:39:39

标签: node.js tensorflow.js sequence-to-sequence

嗨,我正在尝试使用tensorflow.js中的序列2序列模型构建文本摘要器。

我的数据集(示例):

{
Text: i want to return this product because it was broken when i received it.
Expected Output: broken Product.
}

我可以使用预先训练的word2vec模型将文本转换为单词向量。 我正在将这些词嵌入用于LSTM网络以获得情感价值。

从理论上我可以了解序列2序列模型,但是我无法为编码的想法,即编码器/解码器网络看起来像是真实的代码而困惑。

任何人都可以在tensorflow.js中发布一个简单的编码器和解码器网络示例。

或者如果已经发布了一些示例,您可以在其中分享这些示例。

0 个答案:

没有答案