Datasets documentation
Task templates
Task templates
The tasks supported by Dataset.prepare_for_task() and DatasetDict.prepare_for_task().
class datasets.AutomaticSpeechRecognition
< source >( task: str = 'automatic-speech-recognition'audio_column: str = 'audio'transcription_column: str = 'transcription' )
class datasets.AudioClassification
< source >( task: str = 'audio-classification'audio_column: str = 'audio'label_column: str = 'labels' )
class datasets.ImageClassification
< source >( task: str = 'image-classification'image_column: str = 'image'label_column: str = 'labels' )
class datasets.LanguageModeling
< source >( task: str = 'language-modeling'text_column: str = 'text' )
class datasets.QuestionAnsweringExtractive
< source >( task: str = 'question-answering-extractive'question_column: str = 'question'context_column: str = 'context'answers_column: str = 'answers' )
class datasets.Summarization
< source >( task: str = 'summarization'text_column: str = 'text'summary_column: str = 'summary' )
class datasets.TextClassification
< source >( task: str = 'text-classification'text_column: str = 'text'label_column: str = 'labels' )