Coarse training
Fit
-
void eddl::fit(model m, const vector<Tensor*> &in, const vector<Tensor*> &out, int batch, int epochs)
Trains the model for a fixed number of epochs (iterations on a dataset).
- Parameters
m – Model to train
in – Input data (features)
out – Output data (labels)
batch – Number of samples per gradient update
epochs – Number of epochs to train the model. An epoch is an iteration over the entire data provided
- Returns
(void) Trains the model
Example:
model net = Model({in}, {out});
// Build model
...
// Train model
fit(net, {x_train}, {y_train}, batch_size, epochs);