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
Age
Anomaly detection
Audio
accentdbcommon_voicecrema_ddementiabankfussgroovegtzangtzan_music_speechlibrispeechlibrittsljspeechnsynthsaveespeech_commandsspoken_digittedliumuser_libri_audiovctkvoxcelebvoxforgextreme_syes_no
Biology
Categorical
Common sense reasoning
Computer science
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
Dialog act labeling
Dialogue
Document summarization
Facial attributes
Fine grained image classification
Gender
Graph
Graphs
Health
Image
abstract_reasoningaflw2k3dai2dcaptionbccdbeansbee_datasetbigearthnetbinarized_mnistbinary_alpha_digitscaltech101celeb_aceleb_a_hqcityscapesclevrcliccoil100covrdiv2kdownsampled_imagenetdspritesflicimagenet2012imagenet2012_corruptedimagenet2012_fewshotimagenet2012_multilabelimagenet2012_realimagenet2012_subsetimagenet_aimagenet_ltimagenet_piimagenet_rimagenet_resizedimagenet_sketchimagenet_v2imagenetteimagewangkittilfwlost_and_foundlsunlvismalarianyu_depth_v2open_images_challenge2019_detectionopen_images_v4oxford_flowers102oxford_iiit_petpasspatch_camelyonpet_finderplaces365_smallplacesfullplant_leavesplant_villageplantae_kpneumonia_mnistquickdraw_bitmapref_cocoresisc45robomimic_mgrobomimic_mhrobomimic_phrock_paper_scissorss3o4dscene_parse150shapes3dsiscoresmallnorbso2satstanford_dogsstanford_online_productsstl10sun397svhn_croppedsymmetric_solidstf_flowersthe300w_lpwake_vision
Image classification
abstract_reasoningbigearthnetcaltech101caltech_birds2010caltech_birds2011cars196cassavacats_vs_dogsceleb_achexpertcifar10cifar100cifar100_ncifar10_1cifar10_corruptedcifar10_hcifar10_ncitrus_leavescmaterdbcolorectal_histologycolorectal_histology_largecontrolled_noisy_web_labelscurated_breast_imaging_ddsmcycle_gandeep_weedsdiabetic_retinopathy_detectiondmlabdomainnetdtdemnisteurosatfashion_mnistflicfood101geirhos_conflict_stimulihorses_or_humansi_naturalist2017i_naturalist2018i_naturalist2021imagenet2012imagenet2012_subsetimagenet_resizedimagenet_sketchimagenetteimagewangkittikmnistlvismalariamnistmnist_corruptedomniglotopen_images_v4oxford_flowers102oxford_iiit_petplaces365_smallplant_villagepneumonia_mnistresisc45siscoresmallnorbstanford_dogsstanford_online_productsstl10sun397svhn_croppeduc_mercedvisual_domain_decathlonwake_vision
Image clustering
Image compression
Image generation
binarized_mnistceleb_aceleb_a_hqcityscapesclevrimagenet2012imagenet2012_subsetoxford_flowers102stanford_dogsstl10
Image segmentation
Image super resolution
Image to image translation
Instance segmentation
Language modeling
big_patentbillsumblimpdatabricks_dollydicesdolmae2e_cleanedirc_disentanglementlambadalibrispeech_lmlm1bmath_qaopuspaws_wikipaws_x_wikipg19redditsamsumsnlisquadtedlium
Linguistic acceptability
Machine translation
Monolingual
ag_news_subsetai2_arc_with_irarcbeirbooksumbool_qbot_adversarial_dialoguecovrdicesdolmae2e_cleanedimdb_reviewskittilambadalibrispeechlibrispeech_lmlibrittsljspeechlm1bnatural_questionsnatural_questions_openopenbookqapaws_wikiplant_villagequacracereal_toxicity_promptsredditsaveeschema_guided_dialoguesci_tailscicitescientific_paperssentiment140snlispeech_commandsspoken_digitsquadstory_clozetedliumtrectrivia_qa
Movies and tv shows
Multilingual
Natural language inference
Natural language understanding
ag_news_subsetanlibeirbool_qclevrcovrdatabricks_dollydicesimdb_reviewsmath_datasetmath_qamlqamulti_newsnatural_instructionsnatural_questionsnatural_questions_openopenbookqaopuspaws_wikipaws_x_wikipg19piqaqascquacraceschema_guided_dialoguesci_tailsentiment140snlisquadstory_clozetrectrivia_qa
Nearest neighbors
News
Object detection
Open domain question answering
Out of distribution detection
Question answering
beirbool_qclevrcoqacosmos_qadatabricks_dollymath_datasetmath_qamlqanatural_instructionsnatural_questionsnatural_questions_openopenbookqapiqaqascquacracesquadstory_clozetrivia_qatydi_qaweb_questionsxquad
Question generation
Race national ethnic origin
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
aloha_mobileasimov_dilemmas_auto_valasimov_dilemmas_scifi_trainasimov_dilemmas_scifi_valasimov_injury_valasimov_multimodal_auto_valasimov_multimodal_manual_valasimov_v2_constraints_with_rationaleasimov_v2_constraints_without_rationaleasimov_v2_injuriesasimov_v2_videosasu_table_top_converted_externally_to_rldsaustin_buds_dataset_converted_externally_to_rldsaustin_sailor_dataset_converted_externally_to_rldsaustin_sirius_dataset_converted_externally_to_rldsbc_zberkeley_autolab_ur5berkeley_cable_routingberkeley_fanuc_manipulationberkeley_gnm_cory_hallberkeley_gnm_reconberkeley_gnm_sac_sonberkeley_mvp_converted_externally_to_rldsberkeley_rpt_converted_externally_to_rldsbridgebridge_data_msrcmu_franka_exploration_dataset_converted_externally_to_rldscmu_play_fusioncmu_stretchcolumbia_cairlab_pusht_realconq_hose_manipulationdlr_edan_shared_control_converted_externally_to_rldsdlr_sara_grid_clamp_converted_externally_to_rldsdlr_sara_pour_converted_externally_to_rldsdobbeeth_agent_affordancesfmbfractal20220817_dataiamlab_cmu_pickup_insert_converted_externally_to_rldsimperialcollege_sawyer_wrist_camio_ai_techjaco_playkaist_nonprehensile_converted_externally_to_rldskukamaniskill_dataset_converted_externally_to_rldsmimic_playmt_optnyu_door_opening_surprising_effectivenessnyu_franka_play_dataset_converted_externally_to_rldsnyu_rot_dataset_converted_externally_to_rldsplex_robosuiterobo_setroboturkspoc_robotstanford_hydra_dataset_converted_externally_to_rldsstanford_kuka_multimodal_dataset_converted_externally_to_rldsstanford_mask_vit_converted_externally_to_rldsstanford_robocook_converted_externally_to_rldstaco_playtidybottokyo_u_lsmo_converted_externally_to_rldstotoucsd_kitchen_dataset_converted_externally_to_rldsucsd_pick_and_place_dataset_converted_externally_to_rldsuiuc_d3fieldusc_cloth_sim_converted_externally_to_rldsutaustin_mutexutokyo_pr2_opening_fridge_converted_externally_to_rldsutokyo_pr2_tabletop_manipulation_converted_externally_to_rldsutokyo_saytap_converted_externally_to_rldsutokyo_xarm_bimanual_converted_externally_to_rldsutokyo_xarm_pick_and_place_converted_externally_to_rldsvima_converted_externally_to_rldsviola
Scene classification
Semantic segmentation
bigearthnetcityscapeskittilost_and_foundnyu_depth_v2open_images_v4places365_smallref_cocoscene_parse150segment_anythingso2sat
Sentiment analysis
Sequence modeling
Sequence to sequence language modeling
Speech
Speech recognition
Structured
cherry_blossomscovid19cs_restaurantsdartdiamondsforest_firesgenomics_oodgerman_credit_numerichiggshowellirismovie_lensmovielensweb_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_qbot_adversarial_dialoguebuccc4c4_wsrscaltech101cfqcityscapescivil_commentsclevrclinc_oosconll2002conll2003corr2causecos_ecovrdatabricks_dollydefinite_pronoun_resolutiondicesdoc_nlidolmadolphin_number_worddropdspritese2e_cleanederaser_multi_rcesnliflicgapgemgluegoemotionsgpt3gsm8khellaswagimdb_reviewsirc_disentanglementkittilambadalfwlibrispeechlibrispeech_lmlibrittsljspeechlm1blost_and_foundmath_datasetmath_qamctacomedia_summlqamovie_rationalesmrqamulti_newsmulti_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_ccqa4mreqascquacqualityracereal_toxicity_promptsreddit_disentanglementreddit_tifuref_cocoresisc45robonetrock_yousalient_span_wikipediasamsumscanschema_guided_dialoguesci_tailscicitescientific_papersscrollssentiment140smallnorbsnlispoken_digitsquadsquad_question_generationstanford_dogsstar_cfqstory_clozesummscreensun397super_gluesvhn_croppedtatoebated_hrlr_translateted_multi_translatetedliumtiny_shakespearetrectrivia_qaunified_qauniversal_dependenciesunnatural_instructionsuser_libri_textwebvidwiki40bwiki_dialogwikiannwikipediawikipedia_toxicity_subtypeswinograndewordnetwsc273xnlixtreme_pawsxxtreme_xnliyelp_polarity_reviews
Text classification
ag_news_subsetbool_qbot_adversarial_dialoguedicesimdb_reviewsnatural_instructionspaws_wikipaws_x_wikisentiment140trec
Text classification toxicity prediction
Text generation
aeslcbig_patentbillsumbooksumbool_qdatabricks_dollye2e_cleanedlambadalm1bmath_qamctaconatural_instructionsnatural_questionsnewsroomopenbookqapiqaracereal_toxicity_promptsredditreddit_tifusamsumschema_guided_dialoguescientific_paperssquadstory_clozetrivia_qa
Text simplification
Text summarization
aeslcbig_patentbillsumbooksumdatabricks_dollymulti_newsnewsroomredditreddit_tifusamsumscientific_papers
Time series
Token classification
Tracking
Trajectory
Translate
floresmtntwmt13_translatewmt14_translatewmt15_translatewmt16_translatewmt17_translatewmt18_translatewmt19_translatewmt_t2t_translate
Uncategorized
Unsupervised anomaly detection
Video
abstract_reasoningbair_robot_pushing_smalldavisflicmoving_mnistrobonetstarcraft_videotaoucf101webvidyoutube_vis