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

Data loader for SST2 and MNL (GLUE tasks).

Classes

ClassifierDataProcessorConfig

ClassifierDataset

Base class for dataset that load their raw data from TSV files.

DataProcessor

Base class for processors that load their raw data from TFDS.

MNLIDataProcessor

The data processor responsible for creating the MNLI dataloader instance.

MNLIDataProcessorConfig

MNLIDataset

SST2 dataset processor for sentiment analysis.

SST2DataProcessor

The data processor responsible for creating the SST2 dataloader instance.

SST2DataProcessorConfig

SST2Dataset

SST2 dataset processor for sentiment analysis.