Module redlite.dataset.memory_dataset

Classes

MemoryDataset

class MemoryDataset(
    *,
    data: collections.abc.Iterator[redlite._core.DatasetItem],
    name: str,
    split: Literal['test', 'train'] = 'test',
    labels: dict[str, str] | None = None
)

Dataset from an iterator returning DatasetItems.

Data iterator will be greedily executed and all items held in an array.

• data (Iterator[DatasetItem]): Iterator returning data points.
• name (str): Dataset name.
• split (str): Dataset split. Defaults to "test".
• labels (dict[str, str]): Labels. Defaults to empty dict.

Ancestors (in MRO)

• redlite.dataset._load.ValidatingDataset
• redlite._core.NamedDataset
• collections.abc.Sized
• collections.abc.Iterable

Descendants

• redlite.dataset.json_dataset.JSONDataset