de.datexis.parvec.encoder

Class LabelSeeker

java.lang.Object

de.datexis.parvec.encoder.LabelSeeker

public class LabelSeeker
extends Object

This is primitive seeker for nearest labels. It's used instead of basic wordsNearest method because for ParagraphVectors only labels should be taken into account, not individual words

Author:

raver119@gmail.com

Constructor Summary

Constructors

Constructor and Description
LabelSeeker(List<String> labelsUsed, org.deeplearning4j.models.embeddings.inmemory.InMemoryLookupTable<org.deeplearning4j.models.word2vec.VocabWord> lookupTable)

Method Summary

Modifier and Type	Method and Description
List<org.nd4j.linalg.primitives.Pair<String,Double>>	getScores(org.nd4j.linalg.api.ndarray.INDArray vector) This method accepts vector, that represents any document, and returns distances between this document, and previously trained categories
org.nd4j.linalg.api.ndarray.INDArray	getScoresAsVector(org.nd4j.linalg.api.ndarray.INDArray vector)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

LabelSeeker

public LabelSeeker(List<String> labelsUsed,
                   org.deeplearning4j.models.embeddings.inmemory.InMemoryLookupTable<org.deeplearning4j.models.word2vec.VocabWord> lookupTable)

Method Detail

getScores

public List<org.nd4j.linalg.primitives.Pair<String,Double>> getScores(org.nd4j.linalg.api.ndarray.INDArray vector)

This method accepts vector, that represents any document, and returns distances between this document, and previously trained categories

Returns:

getScoresAsVector

public org.nd4j.linalg.api.ndarray.INDArray getScoresAsVector(org.nd4j.linalg.api.ndarray.INDArray vector)