public class SparkSkipGram extends BaseSparkLearningAlgorithm
Modifier and Type | Field and Description |
---|---|
protected AtomicLong |
counter |
protected TrainingDriver<SkipGramRequestMessage> |
driver |
protected ThreadLocal<Frame<SkipGramRequestMessage>> |
frame |
nextRandom, vectorsConfiguration, vocabCache
Constructor and Description |
---|
SparkSkipGram() |
Modifier and Type | Method and Description |
---|---|
Frame<? extends TrainingMessage> |
frameSequence(Sequence<ShallowSequenceElement> sequence,
AtomicLong nextRandom,
double learningRate) |
String |
getCodeName() |
TrainingDriver<? extends TrainingMessage> |
getTrainingDriver() |
protected void |
iterateSample(ShallowSequenceElement word,
ShallowSequenceElement lastWord,
AtomicLong nextRandom,
double lr) |
double |
learnSequence(Sequence<ShallowSequenceElement> sequence,
AtomicLong nextRandom,
double learningRate,
BatchSequences<ShallowSequenceElement> batchSequences) |
applySubsampling, configure, finish, isEarlyTerminationHit, learnSequence, pretrain
protected transient AtomicLong counter
protected transient ThreadLocal<Frame<SkipGramRequestMessage>> frame
protected TrainingDriver<SkipGramRequestMessage> driver
public String getCodeName()
public double learnSequence(Sequence<ShallowSequenceElement> sequence, AtomicLong nextRandom, double learningRate, BatchSequences<ShallowSequenceElement> batchSequences)
public Frame<? extends TrainingMessage> frameSequence(Sequence<ShallowSequenceElement> sequence, AtomicLong nextRandom, double learningRate)
protected void iterateSample(ShallowSequenceElement word, ShallowSequenceElement lastWord, AtomicLong nextRandom, double lr)
public TrainingDriver<? extends TrainingMessage> getTrainingDriver()
Copyright © 2020. All rights reserved.