org.deeplearning4j.spark.models.embeddings.word2vec

Class VocabHolder

java.lang.Object

org.deeplearning4j.spark.models.embeddings.word2vec.VocabHolder

All Implemented Interfaces:

Serializable

public class VocabHolder
extends Object
implements Serializable

Author:

raver119@gmail.com

See Also:

Serialized Form

Method Summary

Modifier and Type	Method and Description
static VocabHolder	getInstance()
Iterable<Map.Entry<VocabWord,org.nd4j.linalg.api.ndarray.INDArray>>	getSplit(VocabCache<VocabWord> vocabCache)
org.nd4j.linalg.api.ndarray.INDArray	getSyn0Vector(Integer wordIndex, VocabCache<VocabWord> vocabCache)
org.nd4j.linalg.api.ndarray.INDArray	getSyn1Vector(Integer point)
void	setSeed(long seed, int vectorLength)

Methods inherited from class java.lang.Object

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

Method Detail

getInstance

public static VocabHolder getInstance()

setSeed

public void setSeed(long seed,
                    int vectorLength)

getSyn0Vector

public org.nd4j.linalg.api.ndarray.INDArray getSyn0Vector(Integer wordIndex,
                                                          VocabCache<VocabWord> vocabCache)

getSyn1Vector

public org.nd4j.linalg.api.ndarray.INDArray getSyn1Vector(Integer point)

getSplit

public Iterable<Map.Entry<VocabWord,org.nd4j.linalg.api.ndarray.INDArray>> getSplit(VocabCache<VocabWord> vocabCache)