Example usage for org.deeplearning4j.models.embeddings.wordvectors WordVectors wordsNearest

Introduction

In this page you can find the example usage for org.deeplearning4j.models.embeddings.wordvectors WordVectors wordsNearest.

Prototype

Collection<String> wordsNearest(Collection<String> positive, Collection<String> negative, int top);

Source Link

Document

Words nearest based on positive and negative words

Usage

From source file:edu.polyu.comp5412.word2vec.Word2VecTestChi.java

public static void main(String[] args) throws Exception {
    WordVectors vec = WordVectorSerializer.loadTxtVectors(new File("poem-vec.txt"));

    Collection<String> words = vec.wordsNearest(Arrays.asList("", ""), Arrays.asList(""), 5);
    System.out.println(words);/*from ww  w.j a  va2s  .co  m*/

    words = vec.wordsNearest(Arrays.asList("", ""), Arrays.asList(""), 5);
    System.out.println(words);

    String[] testwords = new String[] { "", "", "", "", "" };
    for (String s : testwords) {
        Collection<String> lst = vec.wordsNearest(s, 5);
        List<SimWord> simwords = new ArrayList();
        for (String w : lst) {
            SimWord sw = new SimWord();
            sw.word = w;
            sw.similarlity = vec.similarity(s, w);
            simwords.add(sw);
        }
        System.out.println(s);
        System.out.println("=====================");
        for (SimWord sw : simwords) {
            System.out.println(sw.word + "\t" + sw.similarlity);
        }
    }
}

---