cerebras.modelzoo.data.nlp.bert.BertClassifierDataProcessor.DataProcessor#

class cerebras.modelzoo.data.nlp.bert.BertClassifierDataProcessor.DataProcessor(config)[source]#

Bases: abc.ABC

Base class for processors that load their raw data from TFDS. Child classes must provide map_fn, name.

Methods

create_dataloader

create_dataset

previous

cerebras.modelzoo.data.nlp.bert.BertClassifierDataProcessor.ClassifierDataset

next

cerebras.modelzoo.data.nlp.bert.BertClassifierDataProcessor.MNLIDataProcessor