public class SGD<MODEL,DATATYPE,PTYPE extends Parameters<PTYPE>> extends Object
Modifier and Type | Field and Description |
---|---|
int |
batchSize |
DifferentiableObjectiveFunction<MODEL,DATATYPE,PTYPE> |
fcn |
LearningRate<PTYPE> |
learningRate |
int |
maxEpochs |
MODEL |
model |
Constructor and Description |
---|
SGD() |
Modifier and Type | Method and Description |
---|---|
static void |
main(String[] args) |
void |
train(DataSource<DATATYPE> data) |
double |
value(MODEL model,
DATATYPE data) |
public int maxEpochs
public int batchSize
public LearningRate<PTYPE extends Parameters<PTYPE>> learningRate
public DifferentiableObjectiveFunction<MODEL,DATATYPE,PTYPE extends Parameters<PTYPE>> fcn
public SGD()