public class SparkCBOW extends BaseSparkLearningAlgorithm
Modifier and Type | Field and Description |
---|---|
protected ThreadLocal<Frame<CbowRequestMessage>> |
frame |
nextRandom, vectorsConfiguration, vocabCache
Constructor and Description |
---|
SparkCBOW() |
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 currentWord,
int[] windowWords,
AtomicLong nextRandom,
double alpha,
boolean isInference,
int numLabels,
boolean trainWords,
INDArray inferenceVector) |
double |
learnSequence(Sequence<ShallowSequenceElement> sequence,
AtomicLong nextRandom,
double learningRate,
BatchSequences<ShallowSequenceElement> batchSequences) |
applySubsampling, configure, finish, isEarlyTerminationHit, learnSequence, pretrain
protected transient ThreadLocal<Frame<CbowRequestMessage>> frame
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 currentWord, int[] windowWords, AtomicLong nextRandom, double alpha, boolean isInference, int numLabels, boolean trainWords, INDArray inferenceVector)
public TrainingDriver<? extends TrainingMessage> getTrainingDriver()
Copyright © 2020. All rights reserved.