Package org.deeplearning4j.models.embeddings.learning

Interface ElementsLearningAlgorithm<T extends SequenceElement>

All Known Implementing Classes:

CBOW, SkipGram

public interface ElementsLearningAlgorithm<T extends SequenceElement>

Method Summary

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

Method Detail

getCodeName

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,
                     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

learnSequence

double learnSequence(Sequence<T> sequence,
                     AtomicLong nextRandom,
                     double learningRate,
                     BatchSequences<T> batchSequences)

isEarlyTerminationHit

boolean isEarlyTerminationHit()

finish

void finish()