public class Word2VecParametersV3 extends ModelParametersSchemaV3
| Modifier and Type | Field and Description |
|---|---|
int |
epochs
Number of training iterations to run
|
float |
initLearningRate
Set the starting learning rate
|
int |
minWordFreq
This will discard words that appear less than
|
Word2VecNormModel |
normModel
Use Hierarchical Softmax
|
float |
sentSampleRate
Set threshold for occurrence of words.
|
int |
vecSize
Set size of word vectors
|
int |
windowSize
Set max skip length between words
|
Word2VecWordModel |
wordModel
Use the Skip-Gram model
|
categoricalEncoding, checkpoint, distribution, foldAssignment, foldColumn, huberAlpha, ignoreConstCols, ignoredColumns, keepCrossValidationFoldAssignment, keepCrossValidationPredictions, maxRuntimeSecs, modelId, nfolds, offsetColumn, parallelizeCrossValidation, quantileAlpha, responseColumn, scoreEachIteration, stoppingMetric, stoppingRounds, stoppingTolerance, trainingFrame, tweediePower, validationFrame, weightsColumn| Constructor and Description |
|---|
Word2VecParametersV3()
Public constructor
|
| Modifier and Type | Method and Description |
|---|---|
java.lang.String |
toString()
Return the contents of this object as a JSON String.
|
public int vecSize
public int windowSize
public float sentSampleRate
public Word2VecNormModel normModel
public int epochs
public int minWordFreq
public float initLearningRate
public Word2VecWordModel wordModel
public java.lang.String toString()
toString in class ModelParametersSchemaV3