Dataset and DataLoader

[fszewczyk]

Implement an abstraction of a Dataset and a DataLoader that can perform batching.

Usage

// This is our XOR dataset. It maps from Vec32 to Vec32
Dataset<Vec32, Vec32> data;
data.addSample(Vec32::of(0, 0), Vec32::of(0));
data.addSample(Vec32::of(0, 1), Vec32::of(1));
data.addSample(Vec32::of(1, 0), Vec32::of(1));
data.addSample(Vec32::of(1, 1), Vec32::of(0));

// The is the data loader, it will take care of batching
size_t batchSize = 2;
bool shuffle = true;
DataLoader loader(data, batchSize, shuffle);

// DataLoader deals internally with generating batches
for (const auto &[in, out] : loader){
    // in and out are of type std::vector<Vec32>
    mlp -> forward(in);
    (...)
}