org.deeplearning4j.spark.models.sequencevectors.learning.sequence

Class SparkDM

java.lang.Object

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

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

org.deeplearning4j.spark.models.sequencevectors.learning.sequence.SparkDM

All Implemented Interfaces:

ElementsLearningAlgorithm<ShallowSequenceElement>, SparkElementsLearningAlgorithm

public class SparkDM
extends SparkCBOW

Spark implementation for PV-DM training algorithm

Field Summary

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

frame

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

nextRandom, vectorsConfiguration, vocabCache

Constructor Summary

Constructors

Constructor and Description
SparkDM()

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()

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

iterateSample

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

Constructor Detail

SparkDM

public SparkDM()

Method Detail

getCodeName

public java.lang.String getCodeName()

Specified by:

getCodeName in interface ElementsLearningAlgorithm<ShallowSequenceElement>

Overrides:

getCodeName in class SparkCBOW

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)

Specified by:

frameSequence in interface SparkElementsLearningAlgorithm

Overrides:

frameSequence in class SparkCBOW

getTrainingDriver

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

Specified by:

getTrainingDriver in interface SparkElementsLearningAlgorithm

Overrides:

getTrainingDriver in class SparkCBOW