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

Class BaseSparkSequenceLearningAlgorithm

java.lang.Object

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

All Implemented Interfaces:

SequenceLearningAlgorithm<ShallowSequenceElement>, SparkSequenceLearningAlgorithm

public abstract class BaseSparkSequenceLearningAlgorithm
extends java.lang.Object
implements SparkSequenceLearningAlgorithm

Field Summary

Fields

Modifier and Type	Field and Description
protected ElementsLearningAlgorithm<ShallowSequenceElement>	elementsLearningAlgorithm
protected VectorsConfiguration	vectorsConfiguration
protected VocabCache<ShallowSequenceElement>	vocabCache

Constructor Summary

Constructors

Constructor and Description
BaseSparkSequenceLearningAlgorithm()

Method Summary

Modifier and Type	Method and Description
void	configure(VocabCache<ShallowSequenceElement> vocabCache, WeightLookupTable<ShallowSequenceElement> lookupTable, VectorsConfiguration configuration)
void	finish()
ElementsLearningAlgorithm<ShallowSequenceElement>	getElementsLearningAlgorithm()
org.nd4j.linalg.api.ndarray.INDArray	inferSequence(Sequence<ShallowSequenceElement> sequence, long nextRandom, double learningRate, double minLearningRate, int iterations) This method does training on previously unseen paragraph, and returns inferred vector
boolean	isEarlyTerminationHit()
void	pretrain(SequenceIterator<ShallowSequenceElement> iterator)

Methods inherited from class java.lang.Object

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

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

getCodeName, learnSequence

Field Detail

vocabCache

protected transient VocabCache<ShallowSequenceElement> vocabCache

vectorsConfiguration

protected transient VectorsConfiguration vectorsConfiguration

elementsLearningAlgorithm

protected transient ElementsLearningAlgorithm<ShallowSequenceElement> elementsLearningAlgorithm

Constructor Detail

BaseSparkSequenceLearningAlgorithm

public BaseSparkSequenceLearningAlgorithm()

Method Detail

configure

public void configure(VocabCache<ShallowSequenceElement> vocabCache,
                      WeightLookupTable<ShallowSequenceElement> lookupTable,
                      VectorsConfiguration configuration)

Specified by:

configure in interface SequenceLearningAlgorithm<ShallowSequenceElement>

pretrain

public void pretrain(SequenceIterator<ShallowSequenceElement> iterator)

Specified by:

pretrain in interface SequenceLearningAlgorithm<ShallowSequenceElement>

isEarlyTerminationHit

public boolean isEarlyTerminationHit()

Specified by:

isEarlyTerminationHit in interface SequenceLearningAlgorithm<ShallowSequenceElement>

inferSequence

public org.nd4j.linalg.api.ndarray.INDArray inferSequence(Sequence<ShallowSequenceElement> sequence,
                                                          long nextRandom,
                                                          double learningRate,
                                                          double minLearningRate,
                                                          int iterations)

Description copied from interface: SequenceLearningAlgorithm

This method does training on previously unseen paragraph, and returns inferred vector

Specified by:

inferSequence in interface SequenceLearningAlgorithm<ShallowSequenceElement>

Returns:

getElementsLearningAlgorithm

public ElementsLearningAlgorithm<ShallowSequenceElement> getElementsLearningAlgorithm()

Specified by:

getElementsLearningAlgorithm in interface SequenceLearningAlgorithm<ShallowSequenceElement>

finish

public void finish()

Specified by:

finish in interface SequenceLearningAlgorithm<ShallowSequenceElement>