org.deeplearning4j.spark.models.sequencevectors.learning.elements

Class SparkCBOW

java.lang.Object

org.deeplearning4j.spark.models.sequencevectors.learning.elements.BaseSparkLearningAlgorithm

org.deeplearning4j.spark.models.sequencevectors.learning.elements.SparkCBOW

All Implemented Interfaces:

ElementsLearningAlgorithm<ShallowSequenceElement>, SparkElementsLearningAlgorithm

Direct Known Subclasses:

SparkDM

public class SparkCBOW
extends BaseSparkLearningAlgorithm

Field Summary

Fields

Modifier and Type	Field and Description
protected java.lang.ThreadLocal<org.nd4j.parameterserver.distributed.messages.Frame<org.nd4j.parameterserver.distributed.messages.requests.CbowRequestMessage>>	frame

Fields inherited from class org.deeplearning4j.spark.models.sequencevectors.learning.elements.BaseSparkLearningAlgorithm

nextRandom, vectorsConfiguration, vocabCache

Constructor Summary

Constructors

Constructor and Description
SparkCBOW()

Method Summary

Modifier and Type	Method and Description
org.nd4j.parameterserver.distributed.messages.Frame<? extends org.nd4j.parameterserver.distributed.messages.TrainingMessage>	frameSequence(Sequence<ShallowSequenceElement> sequence, java.util.concurrent.atomic.AtomicLong nextRandom, double learningRate)
java.lang.String	getCodeName()
org.nd4j.parameterserver.distributed.training.TrainingDriver<? extends org.nd4j.parameterserver.distributed.messages.TrainingMessage>	getTrainingDriver()
protected void	iterateSample(ShallowSequenceElement currentWord, int[] windowWords, java.util.concurrent.atomic.AtomicLong nextRandom, double alpha, boolean isInference, int numLabels, boolean trainWords, org.nd4j.linalg.api.ndarray.INDArray inferenceVector)

Methods inherited from class org.deeplearning4j.spark.models.sequencevectors.learning.elements.BaseSparkLearningAlgorithm

applySubsampling, configure, finish, isEarlyTerminationHit, learnSequence, pretrain

Methods inherited from class java.lang.Object

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

Field Detail

frame

protected transient java.lang.ThreadLocal<org.nd4j.parameterserver.distributed.messages.Frame<org.nd4j.parameterserver.distributed.messages.requests.CbowRequestMessage>> frame

Constructor Detail

SparkCBOW

public SparkCBOW()

Method Detail

getCodeName

public java.lang.String getCodeName()

frameSequence

public org.nd4j.parameterserver.distributed.messages.Frame<? extends org.nd4j.parameterserver.distributed.messages.TrainingMessage> frameSequence(Sequence<ShallowSequenceElement> sequence,
                                                                                                                                                  java.util.concurrent.atomic.AtomicLong nextRandom,
                                                                                                                                                  double learningRate)

iterateSample

protected void iterateSample(ShallowSequenceElement currentWord,
                             int[] windowWords,
                             java.util.concurrent.atomic.AtomicLong nextRandom,
                             double alpha,
                             boolean isInference,
                             int numLabels,
                             boolean trainWords,
                             org.nd4j.linalg.api.ndarray.INDArray inferenceVector)

getTrainingDriver

public org.nd4j.parameterserver.distributed.training.TrainingDriver<? extends org.nd4j.parameterserver.distributed.messages.TrainingMessage> getTrainingDriver()