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¶