Skip to content

Instantly share code, notes, and snippets.

@tmsimont

tmsimont/Word.java

Created May 9, 2016 19:50
Show Gist options
  • Save tmsimont/293e7065047a3f6222a6f6c94c4caf9c to your computer and use it in GitHub Desktop.
Save tmsimont/293e7065047a3f6222a6f6c94c4caf9c to your computer and use it in GitHub Desktop.
Download ZIP
Raw
Word.java
double foo[] = {0.07662966102361679,-0.15683715045452118,-0.09958696365356445,-0.5582161545753479,0.13371440768241882,0.145623579621315,0.21786752343177795,0.037577372044324875,0.010457501746714115,-0.01822548173367977,0.07133414596319199,0.10989873111248016,-0.004703379701822996,0.24611054360866547,0.1842348575592041,-0.03501710295677185,-0.28120318055152893,0.3167855143547058,-0.021439414471387863,-0.10285429656505585,0.31841185688972473,-0.047598835080862045,-0.4360429048538208,0.19820629060268402,0.02021874487400055,-0.3637917637825012,-0.19628046452999115,0.15301348268985748,-0.001860927906818688,-0.28961265087127686,-0.04005725681781769,-0.28192153573036194,0.12212064117193222,0.21076101064682007,0.006745698396116495,0.09953082352876663,-0.13900607824325562,-0.034233808517456055,0.10835923999547958,0.06281674653291702,0.04705433174967766,0.02800493873655796,-0.03654691204428673,0.0717429369688034,0.21126438677310944,0.061982423067092896,0.015610983595252037,0.18408675491809845,-0.020207442343235016,-0.27804282307624817,0.11301705986261368,0.08801347017288208,0.03192387521266937,-0.1792324334383011,0.04980972781777382,0.22482949495315552,-0.06346631050109863,-0.10168977081775665,-0.019589971750974655,-0.012582260183990002,0.33665329217910767,0.0025205675046890974,-0.009461941197514534,0.3833000659942627,-0.01715477555990219,-0.06369662284851074,0.01154022105038166,0.007030464708805084,-0.0708678588271141,-0.017094450071454048,0.12688760459423065,0.5028663277626038,-0.1583442986011505,-0.28018856048583984,0.1671013981103897,-0.0893111377954483,-0.04531631991267204,0.0687960684299469,0.34074172377586365,-0.09085529297590256,6.943101761862636E-4,0.30315786600112915,0.09421809762716293,-0.2649470269680023,-0.3064609467983246,0.2695615589618683,-0.11702729016542435,-0.3704732060432434,-0.20678825676441193,0.31280359625816345,0.25116559863090515,-0.1313193440437317,-0.11854994297027588,0.17942380905151367,-0.2513428032398224,0.2257017344236374,-0.1161293312907219,0.010692618787288666,-0.09326448291540146,-0.08194870501756668};
wordVectors = WordVectorSerializer.loadTxtVectors(new File("word-vectors.txt"));
Collection<String> lst = wordVectors.wordsNearest(Nd4j.create(foo), 10);
System.out.println(lst);
@tmsimont
Copy link
Author

tmsimont commented May 9, 2016 

Exception in thread "main" java.lang.IllegalArgumentException: Index [1] must not be >= shape[d].
    at org.nd4j.linalg.api.shape.Shape.getOffset(Shape.java:484)
    at org.nd4j.linalg.api.shape.Shape.getDouble(Shape.java:199)
    at org.nd4j.linalg.api.ndarray.BaseNDArray.getDouble(BaseNDArray.java:1407)
    at org.nd4j.linalg.api.ndarray.BaseNDArray.getInt(BaseNDArray.java:1385)
    at org.deeplearning4j.models.embeddings.wordvectors.WordVectorsImpl.wordsNearest(WordVectorsImpl.java:218)
    at TestWord2Vec.test(TestWord2Vec.java:28)
    at TestWord2Vec.main(TestWord2Vec.java:21)
    at sun.reflect.NativeMethodAccessorImpl.invoke0(Native Method)
    at sun.reflect.NativeMethodAccessorImpl.invoke(NativeMethodAccessorImpl.java:57)
    at sun.reflect.DelegatingMethodAccessorImpl.invoke(DelegatingMethodAccessorImpl.java:43)
    at java.lang.reflect.Method.invoke(Method.java:606)
    at com.intellij.rt.execution.application.AppMain.main(AppMain.java:144)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment