Class InMemoryLookupTable.Builder<T extends SequenceElement>
- java.lang.Object
-
- org.deeplearning4j.models.embeddings.inmemory.InMemoryLookupTable.Builder<T>
-
- Enclosing class:
- InMemoryLookupTable<T extends SequenceElement>
public static class InMemoryLookupTable.Builder<T extends SequenceElement> extends Object
-
-
Field Summary
Fields Modifier and Type Field 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 Description Builder()
-
Method Summary
All Methods Instance Methods Concrete Methods Deprecated Methods Modifier and Type Method Description InMemoryLookupTable<T>
build()
InMemoryLookupTable.Builder<T>
cache(@NonNull 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)
-
-
-
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
-
-
Method Detail
-
useHierarchicSoftmax
public InMemoryLookupTable.Builder<T> useHierarchicSoftmax(boolean reallyUse)
-
cache
public InMemoryLookupTable.Builder<T> cache(@NonNull @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()
-
-