org.deeplearning4j.models.embeddings.learning

Interface ElementsLearningAlgorithm<T extends SequenceElement>

All Known Subinterfaces:

SparkElementsLearningAlgorithm

All Known Implementing Classes:

BaseSparkLearningAlgorithm, CBOW, GloVe, SkipGram, SparkCBOW, SparkDBOW, SparkDM, SparkSkipGram

public interface ElementsLearningAlgorithm<T extends SequenceElement>

Implementations of this interface should contain element-related learning algorithms. Like skip-gram, cbow or glove

Method Summary

Modifier and Type	Method and Description
void	configure(VocabCache<T> vocabCache, WeightLookupTable<T> lookupTable, VectorsConfiguration configuration)
void	finish()
java.lang.String	getCodeName()
boolean	isEarlyTerminationHit()
double	learnSequence(Sequence<T> sequence, java.util.concurrent.atomic.AtomicLong nextRandom, double learningRate) This method does training over the sequence of elements passed into it
void	pretrain(SequenceIterator<T> iterator)

Method Detail

getCodeName

java.lang.String getCodeName()

configure

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

pretrain

void pretrain(SequenceIterator<T> iterator)

learnSequence

double learnSequence(Sequence<T> sequence,
                     java.util.concurrent.atomic.AtomicLong nextRandom,
                     double learningRate)

This method does training over the sequence of elements passed into it

Parameters:

sequence -

nextRandom -

learningRate -

Returns:

average score for this sequence

isEarlyTerminationHit

boolean isEarlyTerminationHit()

finish

void finish()