org.deeplearning4j.spark.models.sequencevectors.learning

Interface SparkElementsLearningAlgorithm

All Superinterfaces:

ElementsLearningAlgorithm<ShallowSequenceElement>

All Known Implementing Classes:

BaseSparkLearningAlgorithm, SparkCBOW, SparkDBOW, SparkDM, SparkSkipGram

public interface SparkElementsLearningAlgorithm
extends ElementsLearningAlgorithm<ShallowSequenceElement>

Identification layer for Spark-ready implementations of LearningAlgorithms

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)
org.nd4j.parameterserver.distributed.training.TrainingDriver<? extends org.nd4j.parameterserver.distributed.messages.TrainingMessage>	getTrainingDriver()

Methods inherited from interface org.deeplearning4j.models.embeddings.learning.ElementsLearningAlgorithm

configure, finish, getCodeName, isEarlyTerminationHit, learnSequence, pretrain

Method Detail

getTrainingDriver

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

frameSequence

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)