org.deeplearning4j.models.glove

Class GloveWeightLookupTable.Builder<T extends SequenceElement>

java.lang.Object

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

org.deeplearning4j.models.glove.GloveWeightLookupTable.Builder<T>

Enclosing class:

GloveWeightLookupTable<T extends SequenceElement>

public static class GloveWeightLookupTable.Builder<T extends SequenceElement>
extends InMemoryLookupTable.Builder<T>

Field Summary

Fields inherited from class org.deeplearning4j.models.embeddings.inmemory.InMemoryLookupTable.Builder

gen, lr, negative, seed, useAdaGrad, useHS, vectorLength, vocabCache

Constructor Summary

Constructors

Constructor and Description
Builder()

Method Summary

Modifier and Type	Method and Description
GloveWeightLookupTable<T>	build()
GloveWeightLookupTable.Builder<T>	cache(VocabCache<T> vocab)
GloveWeightLookupTable.Builder<T>	gen(org.nd4j.linalg.api.rng.Random gen)
GloveWeightLookupTable.Builder<T>	lr(double lr) This method is deprecated, since all logic was pulled out from this class
GloveWeightLookupTable.Builder<T>	maxCount(double maxCount)
GloveWeightLookupTable.Builder<T>	negative(double negative)
GloveWeightLookupTable.Builder<T>	seed(long seed)
GloveWeightLookupTable.Builder<T>	useAdaGrad(boolean useAdaGrad)
GloveWeightLookupTable.Builder<T>	vectorLength(int vectorLength)
GloveWeightLookupTable.Builder<T>	xMax(double xMax)

Methods inherited from class org.deeplearning4j.models.embeddings.inmemory.InMemoryLookupTable.Builder

useHierarchicSoftmax

Methods inherited from class java.lang.Object

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

Constructor Detail

Builder

public Builder()

Method Detail

maxCount

public GloveWeightLookupTable.Builder<T> maxCount(double maxCount)

xMax

public GloveWeightLookupTable.Builder<T> xMax(double xMax)

cache

public GloveWeightLookupTable.Builder<T> cache(VocabCache<T> vocab)

Overrides:

cache in class InMemoryLookupTable.Builder<T extends SequenceElement>

negative

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

Overrides:

negative in class InMemoryLookupTable.Builder<T extends SequenceElement>

vectorLength

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

Overrides:

vectorLength in class InMemoryLookupTable.Builder<T extends SequenceElement>

useAdaGrad

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

Overrides:

useAdaGrad in class InMemoryLookupTable.Builder<T extends SequenceElement>

lr

public GloveWeightLookupTable.Builder<T> lr(double lr)

Description copied from class: InMemoryLookupTable.Builder

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

Overrides:

lr in class InMemoryLookupTable.Builder<T extends SequenceElement>

Returns:

gen

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

Overrides:

gen in class InMemoryLookupTable.Builder<T extends SequenceElement>

seed

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

Overrides:

seed in class InMemoryLookupTable.Builder<T extends SequenceElement>

build

public GloveWeightLookupTable<T> build()

Overrides:

build in class InMemoryLookupTable.Builder<T extends SequenceElement>