cerebras.modelzoo.data.nlp.bert.BertClassifierDataProcessor.SST2DataProcessor

class cerebras.modelzoo.data.nlp.bert.BertClassifierDataProcessor.SST2DataProcessor(config)

Bases: cerebras.modelzoo.data.nlp.bert.BertClassifierDataProcessor.DataProcessor

The data processor responsible for creating the SST2 dataloader instance.

Methods

create_dataloader
create_dataset

previous

cerebras.modelzoo.data.nlp.bert.BertClassifierDataProcessor.MNLIDataset

next

cerebras.modelzoo.data.nlp.bert.BertClassifierDataProcessor.SST2DataProcessorConfig