1. public void testDifferentLabels() throws Exception {
    2. ClassPathResource resource = new ClassPathResource("/labeled");
    3. File file = resource.getFile();
    4. LabelAwareSentenceIterator iter = LabelAwareUimaSentenceIterator.createWithPath(file.getAbsolutePath());
    5. ParagraphVectors vec = new ParagraphVectors.Builder()
    6. .minWordFrequency(1).labels(Arrays.asList("negative", "neutral","positive"))
    7. .layerSize(100)
    8. .stopWords(new ArrayList<String>())
    9. vec.fit();
    10. assertNotEquals(vec.lookupTable().vector("UNK"), vec.lookupTable().vector("negative"));
    11. assertNotEquals(vec.lookupTable().vector("UNK"),vec.lookupTable().vector("positive"));