org.deeplearning4j.spark.models.embeddings.word2vec

Class NegativeHolder

java.lang.Object

org.deeplearning4j.spark.models.embeddings.word2vec.NegativeHolder

All Implemented Interfaces:

java.io.Serializable

public class NegativeHolder
extends java.lang.Object
implements java.io.Serializable

Simple singleton holder class for w2v negative sampling, to avoid syn1Neg creation for each spark node

See Also:

Serialized Form

Method Summary

Modifier and Type	Method and Description
static NegativeHolder	getInstance()
void	initHolder(VocabCache<VocabWord> vocabCache, double[] expTable, int layerSize)
protected void	makeTable(int tableSize, double power)

Methods inherited from class java.lang.Object

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

Method Detail

getInstance

public static NegativeHolder getInstance()

initHolder

public void initHolder(@NonNull
                       VocabCache<VocabWord> vocabCache,
                       double[] expTable,
                       int layerSize)

makeTable

protected void makeTable(int tableSize,
                         double power)