Datasets documentation

Task templates

You are viewing main version, which requires installation from source. If you'd like regular pip install, checkout the latest stable version (v2.18.0).
Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

Task templates

The Task API is deprecated in favor of train-eval-index and will be removed in the next major release.

The tasks supported by Dataset.prepare_for_task() and DatasetDict.prepare_for_task().

class datasets.AutomaticSpeechRecognition

< >

( task: str = 'automatic-speech-recognition' audio_column: str = 'audio' transcription_column: str = 'transcription' )

class datasets.AudioClassification

< >

( task: str = 'audio-classification' audio_column: str = 'audio' label_column: str = 'labels' )

class datasets.ImageClassification

< >

( task: str = 'image-classification' image_column: str = 'image' label_column: str = 'labels' )

align_with_features

< >

( features )

class datasets.LanguageModeling

< >

( task: str = 'language-modeling' text_column: str = 'text' )

class datasets.QuestionAnsweringExtractive

< >

( task: str = 'question-answering-extractive' question_column: str = 'question' context_column: str = 'context' answers_column: str = 'answers' )

class datasets.Summarization

< >

( task: str = 'summarization' text_column: str = 'text' summary_column: str = 'summary' )

class datasets.TextClassification

< >

( task: str = 'text-classification' text_column: str = 'text' label_column: str = 'labels' )

align_with_features

< >

( features )