在预测步骤中,输入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 (将#修改为@)