org.deeplearning4j.models.embeddings.inmemory

Class InMemoryLookupTable.Builder<T extends SequenceElement>

java.lang.Object

org.deeplearning4j.models.embeddings.inmemory.InMemoryLookupTable.Builder<T>

Direct Known Subclasses:

GloveWeightLookupTable.Builder

Enclosing class:

InMemoryLookupTable<T extends SequenceElement>

public static class InMemoryLookupTable.Builder<T extends SequenceElement>
extends java.lang.Object

Field Summary

Fields

Modifier and Type	Field and Description
protected org.nd4j.linalg.api.rng.Random	gen
protected double	lr
protected double	negative
protected long	seed
protected boolean	useAdaGrad
protected boolean	useHS
protected int	vectorLength
protected VocabCache<T>	vocabCache

Constructor Summary

Constructors

Constructor and Description
Builder()

Method Summary

Modifier and Type	Method and Description
InMemoryLookupTable<T>	build()
InMemoryLookupTable.Builder<T>	cache(VocabCache<T> vocab)
InMemoryLookupTable.Builder<T>	gen(org.nd4j.linalg.api.rng.Random gen)
InMemoryLookupTable.Builder<T>	lr(double lr) Deprecated.
InMemoryLookupTable.Builder<T>	negative(double negative)
InMemoryLookupTable.Builder<T>	seed(long seed)
InMemoryLookupTable.Builder<T>	useAdaGrad(boolean useAdaGrad)
InMemoryLookupTable.Builder<T>	useHierarchicSoftmax(boolean reallyUse)
InMemoryLookupTable.Builder<T>	vectorLength(int vectorLength)

Methods inherited from class java.lang.Object

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

Field Detail

vectorLength

protected int vectorLength

useAdaGrad

protected boolean useAdaGrad

lr

protected double lr

gen

protected org.nd4j.linalg.api.rng.Random gen

seed

protected long seed

negative

protected double negative

vocabCache

protected VocabCache<T extends SequenceElement> vocabCache

useHS

protected boolean useHS

Constructor Detail

Builder

public Builder()

Method Detail

useHierarchicSoftmax

public InMemoryLookupTable.Builder<T> useHierarchicSoftmax(boolean reallyUse)

cache

public InMemoryLookupTable.Builder<T> cache(@NonNull
                                            VocabCache<T> vocab)

negative

public InMemoryLookupTable.Builder<T> negative(double negative)

vectorLength

public InMemoryLookupTable.Builder<T> vectorLength(int vectorLength)

useAdaGrad

public InMemoryLookupTable.Builder<T> useAdaGrad(boolean useAdaGrad)

lr

@Deprecated
public InMemoryLookupTable.Builder<T> lr(double lr)

Deprecated.

This method is deprecated, since all logic was pulled out from this class

Parameters:

lr -

Returns:

gen

public InMemoryLookupTable.Builder<T> gen(org.nd4j.linalg.api.rng.Random gen)

seed

public InMemoryLookupTable.Builder<T> seed(long seed)

build

public InMemoryLookupTable<T> build()