设计神经网络,可根据游戏反馈调整预训练单词嵌入中的单词向量

时间:2018-07-24 15:04:39

标签: neural-network word2vec word-embedding reward

我正在尝试设计一个神经网络,该神经网络将预先训练的词嵌入的词向量作为输入,并基于游戏反馈输出转换后的词向量。

输入看起来像:

input = [[1.44,-2.44,2.32,.....]] -> there is N words of 300 dimensions for each work.

输出与上面相同,但是尺寸值是根据游戏反馈进行转换的。

当前,我可以构建一个简单的完全连接的层,该层接受如上所述的输入并接受如上所述的输出。但是,我想添加更多功能,例如:

input = [ game state, game action, word_vectors, reward from game]
output = [word_vectors]

但是我不确定如何针对以上内容进行设计。

0 个答案:

没有答案