org.deeplearning4j.models.embeddings.inmemory

Class InMemoryLookupTable.Builder

java.lang.Object

org.deeplearning4j.models.embeddings.inmemory.InMemoryLookupTable.Builder

Direct Known Subclasses:

GloveWeightLookupTable.Builder

Enclosing class:

InMemoryLookupTable

public static class InMemoryLookupTable.Builder
extends Object

Field Summary

Fields

Modifier and Type	Field and Description
protected org.apache.commons.math3.random.RandomGenerator	gen
protected double	lr
protected double	negative
protected long	seed
protected boolean	useAdaGrad
protected int	vectorLength
protected VocabCache	vocabCache

Constructor Summary

Constructors

Constructor and Description
Builder()

Method Summary

Modifier and Type	Method and Description
WeightLookupTable	build()
InMemoryLookupTable.Builder	cache(VocabCache vocab)
InMemoryLookupTable.Builder	gen(org.apache.commons.math3.random.RandomGenerator gen)
InMemoryLookupTable.Builder	lr(double lr)
InMemoryLookupTable.Builder	negative(double negative)
InMemoryLookupTable.Builder	seed(long seed)
InMemoryLookupTable.Builder	useAdaGrad(boolean useAdaGrad)
InMemoryLookupTable.Builder	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.apache.commons.math3.random.RandomGenerator gen

seed

protected long seed

negative

protected double negative

vocabCache

protected VocabCache vocabCache

Constructor Detail

Builder

public Builder()

Method Detail

cache

public InMemoryLookupTable.Builder cache(VocabCache vocab)

negative

public InMemoryLookupTable.Builder negative(double negative)

vectorLength

public InMemoryLookupTable.Builder vectorLength(int vectorLength)

useAdaGrad

public InMemoryLookupTable.Builder useAdaGrad(boolean useAdaGrad)

lr

public InMemoryLookupTable.Builder lr(double lr)

gen

public InMemoryLookupTable.Builder gen(org.apache.commons.math3.random.RandomGenerator gen)

seed

public InMemoryLookupTable.Builder seed(long seed)

build

public WeightLookupTable build()