Task templates

The tasks supported by datasets.Dataset.prepare_for_task and datasets.DatasetDict.prepare_for_task.

class datasets.tasks.AutomaticSpeechRecognition(task: str = 'automatic-speech-recognition', audio_file_path_column: str = 'audio_file_path', transcription_column: str = 'transcription')[source]
class datasets.tasks.ImageClassification(task: str = 'image-classification', image_file_path_column: str = 'image_file_path', label_column: str = 'labels', labels: Union[Tuple[str], NoneType] = None)[source]
class datasets.tasks.QuestionAnsweringExtractive(task: str = 'question-answering-extractive', question_column: str = 'question', context_column: str = 'context', answers_column: str = 'answers')[source]
class datasets.tasks.Summarization(task: str = 'summarization', text_column: str = 'text', summary_column: str = 'summary')[source]
class datasets.tasks.TextClassification(task: str = 'text-classification', text_column: str = 'text', label_column: str = 'labels', labels: Union[Tuple[str], NoneType] = None)[source]