org.deeplearning4j.models.embeddings.learning

Interface ElementsLearningAlgorithm<T extends SequenceElement>

All Known Implementing Classes:

CBOW, GloVe, SkipGram

public interface ElementsLearningAlgorithm<T extends SequenceElement>

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

Author:

[email protected]

Method Summary

Modifier and Type	Method and 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
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

isEarlyTerminationHit

boolean isEarlyTerminationHit()

finish

void finish()