finetuner.tuner.dataset.base module

class finetuner.tuner.dataset.base.BaseSampler(labels, batch_size)[source]

Bases: abc.ABC

property batches
class finetuner.tuner.dataset.base.BaseDataset(*args, **kwds)[source]

Bases: abc.ABC, Generic[finetuner.tuner.dataset.base.AnyLabel]

property labels: List[finetuner.tuner.dataset.base.AnyLabel]

Get the list of labels for all items in the dataset.

Return type

List[~AnyLabel]