2016-06-19 180 views
0

我培训了来自http://deeplearning4j.org/word2vec 的word2vec模型成功,现在在尝试应用 wordsNearest时,此异常:word2vec deeplearning4j ArrayIndexOutOfBoundsException异常

Exception in thread "main" java.lang.ArrayIndexOutOfBoundsException: 99 
at 
org.deeplearning4j.models.embeddings.loader.WordVectorSerializer.loadTxt(WordVectorSerializer.java:1107) 
at 
org.deeplearning4j.models.embeddings.loader.WordVectorSerializer.loadTxtVectors(WordVectorSerializer.java:1033) 
at 
org.deeplearning4j.examples.nlp.word2vec.NearestWords.main(NearestWords.java:13) 
at sun.reflect.NativeMethodAccessorImpl.invoke0(Native Method) 
at 
sun.reflect.NativeMethodAccessorImpl.invoke(NativeMethodAccessorImpl.java:62) 
at 
sun.reflect.DelegatingMethodAccessorImpl.invoke(DelegatingMethodAccessorImpl.java:43) 
at java.lang.reflect.Method.invoke(Method.java:498) 
at com.intellij.rt.execution.application.AppMain.main(AppMain.java:144) 

这是我的代码:

package org.deeplearning4j.examples.nlp.word2vec; 
import java.io.File; 
import java.util.Collection; 
import org.deeplearning4j.models.embeddings.loader.WordVectorSerializer; 
import org.deeplearning4j.models.embeddings.wordvectors.WordVectors; 
public class NearestWords { 
    public static void main(String[] args) throws Exception{ 
     File file = new File("pathToWriteto.txt"); 
     WordVectors vec = WordVectorSerializer.loadTxtVectors(file); 
     Collection<String> similar = vec.wordsNearest("day", 10); 
     System.out.println(similar); 
    } 
} 
+0

请加入Gitter的社区。 https://gitter.im/deeplearning4j/deeplearning4j Skymind的工程师正在那个渠道寻求支持。 – tremstat

回答

0

当前版本(0.4-rc3.10)在loadTxt函数中有一个错误。它在主仓库中得到修复,并将在下一个版本中反映出来。阅读这个github问题你的问题: https://github.com/deeplearning4j/deeplearning4j/issues/1721

现在一个简单的修复方法是将最新的WordVectorSerializer.java复制到您的项目。 https://github.com/deeplearning4j/deeplearning4j/blob/master/deeplearning4j-scaleout/deeplearning4j-nlp/src/main/java/org/deeplearning4j/models/embeddings/loader/WordVectorSerializer.java

相关问题