quaterion.dataset.label_cache_dataset module

class LabelCacheDataset(dataset: IndexingDataset)

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

class LabelCacheDatasetMixin(*args, **kwargs)

Bases: object

load(path)

process_item(index, item)

save(path)

set_mode(mode: LabelCacheMode)

property mode: LabelCacheMode

class LabelCacheIterableDataset(dataset: IndexingIterableDataset)

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

class LabelCacheMode(value)

Bases: Enum

An enumeration.

learn = 1

read = 2

transparent = 0