org.deeplearning4j.spark.models.embeddings.glove

Class GloveParam

java.lang.Object

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

All Implemented Interfaces:

java.io.Serializable

public class GloveParam
extends java.lang.Object
implements java.io.Serializable

See Also:

Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	GloveParam.Builder

Constructor Summary

Constructors

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

Method Summary

Modifier and Type	Method and Description
org.apache.spark.broadcast.Broadcast<CounterMap<java.lang.String,java.lang.String>>	getCoOccurrenceCounts()
org.nd4j.linalg.api.rng.Random	getGen()
double	getLr()
double	getMaxCount()
double	getNegative()
int	getVectorLength()
double	getxMax()
boolean	isUseAdaGrad()
void	setCoOccurrenceCounts(org.apache.spark.broadcast.Broadcast<CounterMap<java.lang.String,java.lang.String>> coOccurrenceCounts)
void	setGen(org.nd4j.linalg.api.rng.Random gen)
void	setLr(double lr)
void	setMaxCount(double maxCount)
void	setNegative(double negative)
void	setUseAdaGrad(boolean useAdaGrad)
void	setVectorLength(int vectorLength)
void	setxMax(double xMax)

Methods inherited from class java.lang.Object

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

Constructor Detail

GloveParam

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

Method Detail

getVectorLength

public int getVectorLength()

setVectorLength

public void setVectorLength(int vectorLength)

isUseAdaGrad

public boolean isUseAdaGrad()

setUseAdaGrad

public void setUseAdaGrad(boolean useAdaGrad)

getLr

public double getLr()

setLr

public void setLr(double lr)

getGen

public org.nd4j.linalg.api.rng.Random getGen()

setGen

public void setGen(org.nd4j.linalg.api.rng.Random gen)

getNegative

public double getNegative()

setNegative

public void setNegative(double negative)

getxMax

public double getxMax()

setxMax

public void setxMax(double xMax)

getMaxCount

public double getMaxCount()

setMaxCount

public void setMaxCount(double maxCount)

getCoOccurrenceCounts

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

setCoOccurrenceCounts

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