public class RNNTrainOptions extends Object implements Serializable
| Modifier and Type | Field and Description |
|---|---|
int |
adagradResetFrequency
How many epochs between resets of the adagrad learning rates.
|
int |
batchSize |
int |
debugOutputSeconds |
int |
epochs
Number of times through all the trees
|
double |
initialAdagradWeight
The value to set the learning rate for each parameter when initializing adagrad.
|
double |
learningRate |
int |
maxTrainTimeSeconds |
double |
regClassification
Regularization cost for the classification matrices
|
double |
regTransform
Regularization cost for the transform matrices and tensors
|
double |
regWordVector
Regularization cost for the word vectors
|
double |
scalingForInit |
| 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(String[] args,
int argIndex) |
String |
toString() |
public int batchSize
public int epochs
public int debugOutputSeconds
public int maxTrainTimeSeconds
public double learningRate
public double scalingForInit
public double regTransform
public double regClassification
public double regWordVector
public double initialAdagradWeight
public int adagradResetFrequency
public double getClassWeight(int i)
public int setOption(String[] args, int argIndex)