public class SparkCBOW extends BaseSparkLearningAlgorithm
| Modifier and Type | Field and Description |
|---|---|
protected java.lang.ThreadLocal<Frame<CbowRequestMessage>> |
frame |
nextRandom, vectorsConfiguration, vocabCache| Constructor and Description |
|---|
SparkCBOW() |
| Modifier and Type | Method and Description |
|---|---|
Frame<? extends TrainingMessage> |
frameSequence(Sequence<ShallowSequenceElement> sequence,
java.util.concurrent.atomic.AtomicLong nextRandom,
double learningRate) |
java.lang.String |
getCodeName() |
TrainingDriver<? extends TrainingMessage> |
getTrainingDriver() |
protected void |
iterateSample(ShallowSequenceElement currentWord,
int[] windowWords,
java.util.concurrent.atomic.AtomicLong nextRandom,
double alpha,
boolean isInference,
int numLabels,
boolean trainWords,
INDArray inferenceVector) |
applySubsampling, configure, finish, isEarlyTerminationHit, learnSequence, pretrainprotected transient java.lang.ThreadLocal<Frame<CbowRequestMessage>> frame
public java.lang.String getCodeName()
public Frame<? extends TrainingMessage> frameSequence(Sequence<ShallowSequenceElement> sequence, java.util.concurrent.atomic.AtomicLong nextRandom, double learningRate)
protected void iterateSample(ShallowSequenceElement currentWord, int[] windowWords, java.util.concurrent.atomic.AtomicLong nextRandom, double alpha, boolean isInference, int numLabels, boolean trainWords, INDArray inferenceVector)
public TrainingDriver<? extends TrainingMessage> getTrainingDriver()