微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

在预测步骤中,输入word2vec#wordsNeeplearning4j中最近的单词列表

如何解决在预测步骤中,输入word2vec#wordsNeeplearning4j中最近的单词列表

我没有一个单词可以预测相似性,而是一个单词列表(句子)。 word2vecdeeplearning4j的所有示例都训练有一个单词列表,但只预测一个单词。

我尝试过:

Indarray wordVector1 = wordVectors.getWordVectorMatrix(sentence);
Collection<String> lst_2 = wordVectors.wordsNearest(wordVector1,10);

其中句子是一个字符串。我也尝试过

Collection<String> lst_2 = wordVectors.wordsNearest(sentence,10);

但是它在第一个nullPointerException和第二个中生成一个空数组,我认为不应。

我看到wordVectors#wordsNearest的另一个签名接受一个“ Indarray”,但是我不知道那是什么,除了那是一个性能的数据结构。

在“尝试仅使用正向量使用wordNearest”片段中也看到了示例here,我尝试了:

Collection<String> lst_2 = wordVectors.wordsNearest(Arrays.asList(sentence.split(" ")),10);

org.deeplearning4j版本= 0.9.1 抱怨该类型。就像一个输入是一个集合一样,需要另一个集合。

有任何提示吗?

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。