Contents

class

SushiAI::Data::Rings

Generates concentric rings labelled by ring index.

Declared in
include/SushiAI/data/dataset.hpp
Inherits
SushiAI::Data::Dataset

Decision boundary is quadratic in inputs, making it linearly non-separable. Distractor features contain pure Gaussian noise.

Public member functions

Rings(SushiBLAS::Engine &engine, RingsOptions options={})

Generates the whole dataset.

Parameters

engine

The engine that allocates the tensors.

options

The problem's shape.

Exceptions

Error

If any extent in options is not positive, or the distractor count is negative.

virtual int64_t batch_size() const noexcept override

Samples per batch.

virtual int64_t input_dim() const noexcept override

Features per sample.

virtual int64_t num_classes() const noexcept override

How many classes the targets span.

virtual std::size_t batch_count() const noexcept override

How many batches an epoch holds.

virtual const Batch & batch(std::size_t index) const override

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 override

A short name for a run report.