public class FlatModelUtils<T extends SequenceElement> extends BasicModelUtils<T>
BasicModelUtils.ArrayComparator, BasicModelUtils.SimilarityComparator, BasicModelUtils.WordSimilarity
CORRECT, EXISTS, lookupTable, normalized, vocabCache, WRONG
Constructor and Description |
---|
FlatModelUtils() |
Modifier and Type | Method and Description |
---|---|
java.util.Collection<java.lang.String> |
wordsNearest(org.nd4j.linalg.api.ndarray.INDArray words,
int top)
This method does full scan against whole vocabulary, building descending list of similar words
|
java.util.Collection<java.lang.String> |
wordsNearest(java.lang.String label,
int n)
This method does full scan against whole vocabulary, building descending list of similar words
|
accuracy, getLabels, init, similarity, similarWordsInVocabTo, wordsNearest, wordsNearestSum, wordsNearestSum, wordsNearestSum
public java.util.Collection<java.lang.String> wordsNearest(java.lang.String label, int n)
wordsNearest
in interface ModelUtils<T extends SequenceElement>
wordsNearest
in class BasicModelUtils<T extends SequenceElement>
label
- n
- public java.util.Collection<java.lang.String> wordsNearest(org.nd4j.linalg.api.ndarray.INDArray words, int top)
wordsNearest
in interface ModelUtils<T extends SequenceElement>
wordsNearest
in class BasicModelUtils<T extends SequenceElement>
words
- top
-