Usage
See our getting-started guide for a quick introduction.
for ex in tfds.load('cifar10', split='train'):
...
All Datasets
Dataset Collections
3d
Abstractive text summarization
Anomaly detection
Audio
accentdbcommon_voicecrema_ddementiabankfussgroovegtzangtzan_music_speechlibrispeechlibrittsljspeechnsynthsaveespeech_commandsspoken_digittedliumuser_libri_audiovctkvoxcelebvoxforgextreme_syes_no
Biology
Categorical
Common sense reasoning
Conditional image generation
Coreference resolution
D4rl
d4rl_adroit_doord4rl_adroit_hammerd4rl_adroit_pend4rl_adroit_relocated4rl_antmazed4rl_mujoco_antd4rl_mujoco_halfcheetahd4rl_mujoco_hopperd4rl_mujoco_walker2d
Density estimation
Dependency parsing
Document summarization
Fine grained image classification
Graph
Graphs
Image
abstract_reasoningaflw2k3dbccdbeansbee_datasetbigearthnetbinarized_mnistbinary_alpha_digitscaltech101celeb_aceleb_a_hqcityscapesclevrcliccoil100div2kdownsampled_imagenetdspritesflicimagenet2012imagenet2012_corruptedimagenet2012_fewshotimagenet2012_multilabelimagenet2012_realimagenet2012_subsetimagenet_aimagenet_ltimagenet_rimagenet_resizedimagenet_sketchimagenet_v2imagenetteimagewangkittilfwlost_and_foundlsunlvismalarianyu_depth_v2open_images_challenge2019_detectionopen_images_v4oxford_flowers102oxford_iiit_petpasspatch_camelyonpet_finderplaces365_smallplacesfullplant_leavesplant_villageplantae_kquickdraw_bitmapref_cocoresisc45rock_paper_scissorss3o4dscene_parse150shapes3dsiscoresmallnorbso2satstanford_dogsstanford_online_productsstl10sun397svhn_croppedsymmetric_solidsthe300w_lp
Image classification
abstract_reasoningbigearthnetcaltech101celeb_aflicimagenet2012imagenet_resizedimagenet_sketchimagenetteimagewangkittilvismalariaopen_images_v4oxford_flowers102oxford_iiit_petplaces365_smallplant_villageresisc45siscoresmallnorbstanford_dogsstanford_online_productsstl10sun397svhn_cropped
Image clustering
Image compression
Image generation
binarized_mnistceleb_aceleb_a_hqcityscapesclevrimagenet2012oxford_flowers102stanford_dogsstl10
Image super resolution
Image to image translation
Image classification
caltech_birds2010caltech_birds2011cars196cassavacats_vs_dogschexpertcifar10cifar100cifar100_ncifar10_1cifar10_corruptedcifar10_ncitrus_leavescmaterdbcolorectal_histologycolorectal_histology_largecontrolled_noisy_web_labelscurated_breast_imaging_ddsmcycle_gandeep_weedsdiabetic_retinopathy_detectiondmlabdomainnetdtdemnisteurosatfashion_mnistfood101geirhos_conflict_stimulihorses_or_humansi_naturalist2017i_naturalist2018i_naturalist2021kmnistmnistmnist_corruptedomniglottf_flowersuc_mercedvisual_domain_decathlon
Instance segmentation
Language modelling
big_patentbillsumblimpe2e_cleanedirc_disentanglementlambadalibrispeech_lmlm1bmath_qaopuspaws_wikipaws_x_wikipg19redditsamsumsnlisquad
Machine translation
Monolingual
ag_news_subsetai2_arc_with_irarcbeirbooksumbool_qe2e_cleanedimdb_reviewskittilambadalibrispeechlibrispeech_lmlibrittsljspeechlm1bnatural_questionsnatural_questions_openopenbookqapaws_wikiplant_villagequacraceredditsaveeschema_guided_dialoguesci_tailscicitescientific_paperssentiment140snlispeech_commandsspoken_digitsquadstory_cloze
Movies and tv shows
Multilingual
Natural language inference
Natural language understanding
ag_news_subsetanlibeirbool_qclevrimdb_reviewsmath_datasetmath_qamlqanatural_instructionsnatural_questionsnatural_questions_openopenbookqaopuspaws_wikipaws_x_wikipg19piqaqascquacraceschema_guided_dialoguesci_tailsentiment140snlisquadstory_cloze
Nearest neighbors
Object detection
Object detection
Open domain question answering
Out of distribution detection
Question answering
beirbool_qclevrmath_datasetmath_qamlqanatural_instructionsnatural_questionsnatural_questions_openopenbookqapiqaqascquacracesquadstory_cloze
Question generation
Question answering
Ranking
Reading comprehension
Recommendation
Reinforcement learning
Rgb d
Rl unplugged
rlu_atarirlu_atari_checkpointsrlu_atari_checkpoints_orderedrlu_control_suiterlu_dmlab_explore_object_rewards_fewrlu_dmlab_explore_object_rewards_manyrlu_dmlab_rooms_select_nonmatching_objectrlu_dmlab_rooms_watermazerlu_dmlab_seekavoid_arena01rlu_locomotionrlu_rwrl
Rlds
Robotics
Scene classification
Semantic segmentation
bigearthnetcityscapeskittilost_and_foundnyu_depth_v2open_images_v4places365_smallref_cocoscene_parse150so2sat
Sentiment analysis
Sequence to sequence language modelling
Speech
Speech recognition
Structured
cherry_blossomscovid19cs_restaurantsdartdiamondsforest_firesgenomics_oodgerman_credit_numerichiggshowellirismovie_lensmovielenstitanicweb_graphweb_nlgwiki_biowiki_table_questionswiki_table_textwine_quality
Summarization
Table to text generation
Tabular
Text
abstract_reasoningaeslcag_news_subsetai2_arcai2_arc_with_iramazon_us_reviewsanlianswer_equivalencearcasqaassetassin2bccdbeirbig_patentbillsumblimpbooksumbool_qbuccc4c4_wsrscaltech101cfqcityscapescivil_commentsclevrclinc_oosconll2002conll2003cos_edefinite_pronoun_resolutiondoc_nlidolphin_number_worddropdspritese2e_cleanederaser_multi_rcesnliflicgapgemgluegoemotionsgpt3gsm8khellaswagimdb_reviewsirc_disentanglementkittilambadalfwlibrispeechlibrispeech_lmlibrittsljspeechlm1blost_and_foundmath_datasetmath_qamctacomedia_summlqamovie_rationalesmrqamulti_nlimulti_nli_mismatchnatural_instructionsnatural_questionsnatural_questions_opennewsroomopen_images_challenge2019_detectionopen_images_v4openbookqaopinion_abstractsopinosisopusoxford_flowers102oxford_iiit_petpara_crawlpatch_camelyonpaws_wikipaws_x_wikipenguinspet_finderpg19piqaplaces365_smallplacesfullplant_leavesplant_villageplantae_kprotein_netq_re_ccqa4mreqascquacqualityracereddit_disentanglementreddit_tifuref_cocoresisc45robonetrock_yousalient_span_wikipediasamsumscanschema_guided_dialoguesci_tailscicitescientific_papersscrollssentiment140smallnorbsnlispoken_digitsquadsquad_question_generationstanford_dogsstar_cfqstory_clozesummscreensun397super_gluesvhn_croppedtiny_shakespearetrecunified_qauniversal_dependenciesunnatural_instructionsuser_libri_textwiki40bwiki_dialogwikiannwikipediawikipedia_toxicity_subtypeswinograndewordnetwsc273xnlixtreme_pawsxxtreme_xnliyelp_polarity_reviews
Text classification
Text generation
aeslcbig_patentbillsumbooksumbool_qe2e_cleanedlambadalm1bmath_qamctaconatural_instructionsnatural_questionsnewsroomopenbookqapiqaraceredditreddit_tifusamsumschema_guided_dialoguescientific_paperssquadstory_cloze
Text summarization
Text simplification
Time series
Token classification
Tracking
Trajectory
Translate
floresmtnttatoebated_hrlr_translateted_multi_translatewmt13_translatewmt14_translatewmt15_translatewmt16_translatewmt17_translatewmt18_translatewmt19_translatewmt_t2t_translate
Uncategorized
Unsupervised anomaly detection
Video
abstract_reasoningbair_robot_pushing_smalldavisflicmoving_mnistrobonetstarcraft_videotaoucf101youtube_vis