org.deeplearning4j.models.embeddings.reader.impl

Class FlatModelUtils<T extends SequenceElement>

java.lang.Object

org.deeplearning4j.models.embeddings.reader.impl.BasicModelUtils<T>

org.deeplearning4j.models.embeddings.reader.impl.FlatModelUtils<T>

All Implemented Interfaces:

ModelUtils<T>

public class FlatModelUtils<T extends SequenceElement>
extends BasicModelUtils<T>

This model reader is suited for model tests, and for cases where flat scan against elements is required. PLEASE NOTE: This reader does NOT normalize underlying weights, it stays intact

Author:

[email protected]

Nested Class Summary

Nested classes/interfaces inherited from class org.deeplearning4j.models.embeddings.reader.impl.BasicModelUtils

BasicModelUtils.ArrayComparator, BasicModelUtils.SimilarityComparator, BasicModelUtils.WordSimilarity

Field Summary

Fields inherited from class org.deeplearning4j.models.embeddings.reader.impl.BasicModelUtils

CORRECT, EXISTS, lookupTable, normalized, vocabCache, WRONG

Constructor Summary

Constructors

Constructor and Description
FlatModelUtils()

Method Summary

Modifier and Type	Method and Description
Collection<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
Collection<String>	wordsNearest(String label, int n) This method does full scan against whole vocabulary, building descending list of similar words

Methods inherited from class org.deeplearning4j.models.embeddings.reader.impl.BasicModelUtils

accuracy, getLabels, init, similarity, similarWordsInVocabTo, wordsNearest, wordsNearestSum, wordsNearestSum, wordsNearestSum

Methods inherited from class java.lang.Object

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

Constructor Detail

FlatModelUtils

public FlatModelUtils()

Method Detail

wordsNearest

public Collection<String> wordsNearest(String label,
                                       int n)

This method does full scan against whole vocabulary, building descending list of similar words

Specified by:

wordsNearest in interface ModelUtils<T extends SequenceElement>

Overrides:

wordsNearest in class BasicModelUtils<T extends SequenceElement>

Parameters:

label -

n -

Returns:

wordsNearest

public Collection<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

Specified by:

wordsNearest in interface ModelUtils<T extends SequenceElement>

Overrides:

wordsNearest in class BasicModelUtils<T extends SequenceElement>

Parameters:

words -

top -

Returns:

the words nearest the mean of the words