Java 输入word2vec#words在预测步骤中显示深入学习4j中的单词列表

我没有一个单词来预测相似性，而是一个单词列表（句子）。

deeplearning4j

中的

word2vec

的所有示例都使用单词列表进行训练，但使用单个单词进行预测

我试过：

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

org.deeplearning4j version=0.9.1正在抱怨该类型。就像一个输入是一个集合一样，需要另一个集合

有什么提示吗

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

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