Skip to content

llmcompressor.transformers.data.cnn_dailymail

Classes:

CNNDailyMailDataset 

CNNDailyMailDataset(
    dataset_args: DatasetArguments,
    split: str,
    processor: Processor,
)

Bases: TextGenerationDataset

Text generation class for the CNN/DailyMail dataset

Parameters:

  • dataset_args (DatasetArguments) –

    configuration settings for dataset loading

  • split (str) –

    split from dataset to load, for instance test or train[:5%]

  • processor (Processor) –

    processor or tokenizer to use on dataset

Source code in src/llmcompressor/transformers/data/cnn_dailymail.py 
23
24
25
26
27
28
29
30
def __init__(
    self, dataset_args: "DatasetArguments", split: str, processor: Processor
):
    dataset_args = deepcopy(dataset_args)
    dataset_args.dataset = "cnn_dailymail"
    dataset_args.dataset_config_name = "3.0.0"

    super().__init__(dataset_args=dataset_args, split=split, processor=processor)