langml.baselines.clf.dataloader

Module Contents

Classes

DataLoader

TFDataLoader

class langml.baselines.clf.dataloader.DataLoader(data: List, tokenizer: object, label2id: Dict, batch_size: int = 32, is_bert: bool = True)[source]

Bases: langml.baselines.BaseDataLoader

__len__(self) → int[source]
static load_data(fpath: str, build_vocab: bool = False) → List[source]
make_iter(self, random: bool = False)[source]
class langml.baselines.clf.dataloader.TFDataLoader(data: List, tokenizer: object, label2id: Dict, batch_size: int = 32, is_bert: bool = True)[source]

Bases: DataLoader

make_iter(self, random: bool = False)[source]
__call__(self, random: bool = False)[source]