org.deeplearning4j.spark.models.embeddings.glove

Class GloveParam.Builder

java.lang.Object

org.deeplearning4j.spark.models.embeddings.glove.GloveParam.Builder

Enclosing class:

GloveParam

public static class GloveParam.Builder
extends java.lang.Object

Constructor Summary

Constructors

Constructor and Description
Builder()

Method Summary

Modifier and Type	Method and Description
GloveParam	build()
GloveParam.Builder	coOccurrenceCounts(org.apache.spark.broadcast.Broadcast<CounterMap<java.lang.String,java.lang.String>> coOccurrenceCounts)
GloveParam.Builder	gen(org.nd4j.linalg.api.rng.Random gen)
GloveParam.Builder	lr(double lr)
GloveParam.Builder	maxCount(double maxCount)
GloveParam.Builder	negative(double negative)
GloveParam.Builder	useAdaGrad(boolean useAdaGrad)
GloveParam.Builder	vectorLength(int vectorLength)
GloveParam.Builder	xMax(double xMax)

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

vectorLength

public GloveParam.Builder vectorLength(int vectorLength)

useAdaGrad

public GloveParam.Builder useAdaGrad(boolean useAdaGrad)

lr

public GloveParam.Builder lr(double lr)

gen

public GloveParam.Builder gen(org.nd4j.linalg.api.rng.Random gen)

negative

public GloveParam.Builder negative(double negative)

xMax

public GloveParam.Builder xMax(double xMax)

maxCount

public GloveParam.Builder maxCount(double maxCount)

coOccurrenceCounts

public GloveParam.Builder coOccurrenceCounts(org.apache.spark.broadcast.Broadcast<CounterMap<java.lang.String,java.lang.String>> coOccurrenceCounts)

build

public GloveParam build()