quaterion.dataset.label_cache_dataset module

class LabelCacheDataset(dataset: IndexingDataset)[source]

Bases: Dataset[Union[SimilarityGroupSample, SimilarityPairSample]], LabelCacheDatasetMixin

class LabelCacheDatasetMixin(*args, **kwargs)[source]

Bases: object

load(path)[source]

process_item(index, item)[source]

save(path)[source]

set_mode(mode: LabelCacheMode)[source]

property mode: LabelCacheMode

class LabelCacheIterableDataset(dataset: IndexingIterableDataset)[source]

Bases: IterableDataset[Union[SimilarityGroupSample, SimilarityPairSample]], LabelCacheDatasetMixin

class LabelCacheMode(value)[source]

Bases: Enum

An enumeration.

learn = 1

read = 2

transparent = 0