public class RNNTrainOptions
extends java.lang.Object
implements java.io.Serializable
Modifier and Type | Field and Description |
---|---|
int |
adagradResetFrequency
How many epochs between resets of the adagrad learning rates.
|
int |
batchSize |
int |
debugOutputEpochs |
int |
epochs
Number of times through all the trees
|
double |
initialAdagradWeight
The value to set the learning rate for each parameter when initializing adagrad.
|
java.lang.String |
initialMatrixLogPath
If set, the initial matrices are logged to this location as a single file
using SentimentModel.toString()
|
double |
learningRate |
int |
maxTrainTimeSeconds |
int |
nThreads |
double |
regClassification
Regularization cost for the classification matrices
|
double |
regTransformMatrix
Regularization cost for the transform matrix
|
double |
regTransformTensor
Regularization cost for the transform tensor
|
double |
regWordVector
Regularization cost for the word vectors
|
double |
scalingForInit |
boolean |
shuffleMatrices
Shuffle matrices when training.
|
Constructor and Description |
---|
RNNTrainOptions() |
Modifier and Type | Method and Description |
---|---|
double |
getClassWeight(int i)
The classWeights can be passed in as a comma separated list of
weights using the -classWeights flag.
|
int |
setOption(java.lang.String[] args,
int argIndex) |
java.lang.String |
toString() |
public int batchSize
public int epochs
public int debugOutputEpochs
public int maxTrainTimeSeconds
public double learningRate
public double scalingForInit
public double regTransformMatrix
public double regClassification
public double regWordVector
public double initialAdagradWeight
public int adagradResetFrequency
public double regTransformTensor
public boolean shuffleMatrices
public java.lang.String initialMatrixLogPath
public int nThreads
public double getClassWeight(int i)
public java.lang.String toString()
toString
in class java.lang.Object
public int setOption(java.lang.String[] args, int argIndex)