org.deeplearning4j.models.embeddings

Interface WeightLookupTable

All Superinterfaces:

Serializable

All Known Implementing Classes:

GloveWeightLookupTable, InMemoryLookupTable

public interface WeightLookupTable
extends Serializable

General weight lookup table

Author:

Adam Gibson

Method Summary

Modifier and Type	Method and Description
void	iterate(VocabWord w1, VocabWord w2) Iterate on the given 2 vocab words
void	iterateSample(VocabWord w1, VocabWord w2, AtomicLong nextRandom, double alpha) Iterate on the given 2 vocab words
int	layerSize() The layer size for the lookup table
org.nd4j.linalg.api.ndarray.INDArray	loadCodes(int[] codes) Loads the co-occurrences for the given codes
void	plotVocab() Render the words via tsne
void	plotVocab(Tsne tsne) Render the words via TSNE
void	putCode(int codeIndex, org.nd4j.linalg.api.ndarray.INDArray code)
void	putVector(String word, org.nd4j.linalg.api.ndarray.INDArray vector) Inserts a word vector
void	resetWeights() Reset the weights of the cache
void	resetWeights(boolean reset) Clear out all weights regardless
void	setLearningRate(double lr) Sets the learning rate
org.nd4j.linalg.api.ndarray.INDArray	vector(String word)
Iterator<org.nd4j.linalg.api.ndarray.INDArray>	vectors() Iterates through all of the vectors in the cache

Method Detail

layerSize

int layerSize()

The layer size for the lookup table

Returns:

the layer size for the lookup table

resetWeights

void resetWeights(boolean reset)

Clear out all weights regardless

Parameters:

reset -

plotVocab

void plotVocab(Tsne tsne)

Render the words via TSNE

Parameters:

tsne - the tsne to use

plotVocab

void plotVocab()

Render the words via tsne

putCode

void putCode(int codeIndex,
             org.nd4j.linalg.api.ndarray.INDArray code)

Parameters:

codeIndex -

code -

loadCodes

org.nd4j.linalg.api.ndarray.INDArray loadCodes(int[] codes)

Loads the co-occurrences for the given codes

Parameters:

codes - the codes to load

Returns:

an ndarray of code.length by layerSize

iterate

void iterate(VocabWord w1,
             VocabWord w2)

Iterate on the given 2 vocab words

Parameters:

w1 - the first word to iterate on

w2 - the second word to iterate on

iterateSample

void iterateSample(VocabWord w1,
                   VocabWord w2,
                   AtomicLong nextRandom,
                   double alpha)

Iterate on the given 2 vocab words

Parameters:

w1 - the first word to iterate on

w2 - the second word to iterate on

nextRandom - nextRandom for sampling

alpha - the alpha to use for learning

putVector

void putVector(String word,
               org.nd4j.linalg.api.ndarray.INDArray vector)

Inserts a word vector

Parameters:

word - the word to insert

vector - the vector to insert

vector

org.nd4j.linalg.api.ndarray.INDArray vector(String word)

Parameters:

word -

Returns:

resetWeights

void resetWeights()

Reset the weights of the cache

setLearningRate

void setLearningRate(double lr)

Sets the learning rate

Parameters:

lr -

vectors

Iterator<org.nd4j.linalg.api.ndarray.INDArray> vectors()

Iterates through all of the vectors in the cache

Returns:

an iterator for all vectors in the cache