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

Class SparkDBOW

java.lang.Object

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

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

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

All Implemented Interfaces:

ElementsLearningAlgorithm<ShallowSequenceElement>, SparkElementsLearningAlgorithm

public class SparkDBOW
extends SparkSkipGram

Spark implementation for PV-DBOW training algorithm

Field Summary

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

counter, driver, frame

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

nextRandom, vectorsConfiguration, vocabCache

Constructor Summary

Constructors

Constructor and Description
SparkDBOW()

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.SparkSkipGram

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

SparkDBOW

public SparkDBOW()

Method Detail

getCodeName

public java.lang.String getCodeName()

Specified by:

getCodeName in interface ElementsLearningAlgorithm<ShallowSequenceElement>

Overrides:

getCodeName in class SparkSkipGram

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 SparkSkipGram

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 SparkSkipGram