Contents

class

SushiAI::Data::Dataset

The seam every training loop reads through.

Declared in
include/SushiAI/data/dataset.hpp

Public member functions

virtual ~Dataset()=default
virtual int64_t batch_size() const noexcept=0

Samples per batch.

virtual int64_t input_dim() const noexcept=0

Features per sample.

virtual int64_t num_classes() const noexcept=0

How many classes the targets span.

virtual std::size_t batch_count() const noexcept=0

How many batches an epoch holds.

virtual const Batch & batch(std::size_t index) const =0

One batch.

Parameters

index

Which batch, in [0, batch_count()).

Returns

The batch; owned by the dataset.

Exceptions

Error

If index is out of range.

virtual std::string_view name() const noexcept=0

A short name for a run report.