@Test public void testWordsNearestBasic1() throws Exception { // WordVectors vec = WordVectorSerializer.loadTxtVectors(new // File("/ext/Temp/Models/model.dat_trans")); vec.setModelUtils(new BasicModelUtils<VocabWord>()); String target = "energy"; INDArray arr1 = vec.getWordVectorMatrix(target).dup(); System.out.println("[-]: " + arr1); System.out.println("[+]: " + Transforms.unitVec(arr1)); Collection<String> list = vec.wordsNearest(target, 10); log.info("Transpose model results:"); printWords(target, list, vec); list = vec.wordsNearest(target, 10); log.info("Transpose model results 2:"); printWords(target, list, vec); list = vec.wordsNearest(target, 10); log.info("Transpose model results 3:"); printWords(target, list, vec); INDArray arr2 = vec.getWordVectorMatrix(target).dup(); assertEquals(arr1, arr2); }
@Test public void testWordsNearestFlat1() throws Exception { vec.setModelUtils(new FlatModelUtils<VocabWord>()); Collection<String> list = vec.wordsNearest("energy", 10); log.info("Flat model results:"); printWords("energy", list, vec); }