TextClassificationInput¶

class flash.text.classification.input.TextClassificationInput(running_stage, *args, transform=None, transform_kwargs=None, input_transforms_registry=None, **kwargs)[source]¶

load_data(hf_dataset, input_key, target_keys=None, target_formatter=None)[source]¶

Loads data into HuggingFace datasets.Dataset.