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
accentdb
common_voice
crema_d
dementiabank
fuss
groove
gtzan
gtzan_music_speech
librispeech
libritts
ljspeech
nsynth
savee
speech_commands
spoken_digit
tedlium
user_libri_audio
vctk
voxceleb
voxforge
xtreme_s
yes_no
Biology
Categorical
Common sense reasoning
Computer science
Conditional image generation
Coreference resolution
D4rl
d4rl_adroit_door
d4rl_adroit_hammer
d4rl_adroit_pen
d4rl_adroit_relocate
d4rl_antmaze
d4rl_mujoco_ant
d4rl_mujoco_halfcheetah
d4rl_mujoco_hopper
d4rl_mujoco_walker2d
Density estimation
Dependency parsing
Dialog act labeling
Dialogue
Document summarization
Facial attributes
Fine grained image classification
Gender
Graph
Graphs
Health
Image
abstract_reasoning
aflw2k3d
bccd
beans
bee_dataset
bigearthnet
binarized_mnist
binary_alpha_digits
caltech101
celeb_a
celeb_a_hq
cityscapes
clevr
clic
coil100
div2k
downsampled_imagenet
dsprites
flic
imagenet2012
imagenet2012_corrupted
imagenet2012_fewshot
imagenet2012_multilabel
imagenet2012_real
imagenet2012_subset
imagenet_a
imagenet_lt
imagenet_pi
imagenet_r
imagenet_resized
imagenet_sketch
imagenet_v2
imagenette
imagewang
kitti
lfw
lost_and_found
lsun
lvis
malaria
nyu_depth_v2
open_images_challenge2019_detection
open_images_v4
oxford_flowers102
oxford_iiit_pet
pass
patch_camelyon
pet_finder
places365_small
placesfull
plant_leaves
plant_village
plantae_k
pneumonia_mnist
quickdraw_bitmap
ref_coco
resisc45
robomimic_mg
robomimic_mh
robomimic_ph
rock_paper_scissors
s3o4d
scene_parse150
shapes3d
siscore
smallnorb
so2sat
stanford_dogs
stanford_online_products
stl10
sun397
svhn_cropped
symmetric_solids
tf_flowers
the300w_lp
wake_vision
Image classification
abstract_reasoning
bigearthnet
caltech101
caltech_birds2010
caltech_birds2011
cars196
cassava
cats_vs_dogs
celeb_a
chexpert
cifar10
cifar100
cifar100_n
cifar10_1
cifar10_corrupted
cifar10_h
cifar10_n
citrus_leaves
cmaterdb
colorectal_histology
colorectal_histology_large
controlled_noisy_web_labels
curated_breast_imaging_ddsm
cycle_gan
deep_weeds
diabetic_retinopathy_detection
dmlab
domainnet
dtd
emnist
eurosat
fashion_mnist
flic
food101
geirhos_conflict_stimuli
horses_or_humans
i_naturalist2017
i_naturalist2018
i_naturalist2021
imagenet2012
imagenet_resized
imagenet_sketch
imagenette
imagewang
kitti
kmnist
lvis
malaria
mnist
mnist_corrupted
omniglot
open_images_v4
oxford_flowers102
oxford_iiit_pet
places365_small
plant_village
pneumonia_mnist
resisc45
siscore
smallnorb
stanford_dogs
stanford_online_products
stl10
sun397
svhn_cropped
uc_merced
visual_domain_decathlon
wake_vision
Image clustering
Image compression
Image generation
binarized_mnist
celeb_a
celeb_a_hq
cityscapes
clevr
imagenet2012
oxford_flowers102
stanford_dogs
stl10
Image segmentation
Image super resolution
Image to image translation
Instance segmentation
Language modeling
big_patent
billsum
blimp
databricks_dolly
dices
dolma
e2e_cleaned
irc_disentanglement
lambada
librispeech_lm
lm1b
math_qa
opus
paws_wiki
paws_x_wiki
pg19
reddit
samsum
snli
squad
tedlium
Linguistic acceptability
Machine translation
Monolingual
ag_news_subset
ai2_arc_with_ir
arc
beir
booksum
bool_q
bot_adversarial_dialogue
dices
dolma
e2e_cleaned
imdb_reviews
kitti
lambada
librispeech
librispeech_lm
libritts
ljspeech
lm1b
natural_questions
natural_questions_open
openbookqa
paws_wiki
plant_village
quac
race
real_toxicity_prompts
reddit
savee
schema_guided_dialogue
sci_tail
scicite
scientific_papers
sentiment140
snli
speech_commands
spoken_digit
squad
story_cloze
tedlium
trec
trivia_qa
Movies and tv shows
Multilingual
Natural language inference
Natural language understanding
ag_news_subset
anli
beir
bool_q
clevr
databricks_dolly
dices
imdb_reviews
math_dataset
math_qa
mlqa
natural_instructions
natural_questions
natural_questions_open
openbookqa
opus
paws_wiki
paws_x_wiki
pg19
piqa
qasc
quac
race
schema_guided_dialogue
sci_tail
sentiment140
snli
squad
story_cloze
trec
trivia_qa
Nearest neighbors
Object detection
Open domain question answering
Out of distribution detection
Question answering
beir
bool_q
clevr
coqa
cosmos_qa
databricks_dolly
math_dataset
math_qa
mlqa
natural_instructions
natural_questions
natural_questions_open
openbookqa
piqa
qasc
quac
race
squad
story_cloze
trivia_qa
tydi_qa
web_questions
xquad
Question generation
Race national ethnic origin
Ranking
Reading comprehension
Recommendation
Reinforcement learning
Rgb d
Rl unplugged
rlu_atari
rlu_atari_checkpoints
rlu_atari_checkpoints_ordered
rlu_control_suite
rlu_dmlab_explore_object_rewards_few
rlu_dmlab_explore_object_rewards_many
rlu_dmlab_rooms_select_nonmatching_object
rlu_dmlab_rooms_watermaze
rlu_dmlab_seekavoid_arena01
rlu_locomotion
rlu_rwrl
Rlds
Robotics
aloha_mobile
asu_table_top_converted_externally_to_rlds
austin_buds_dataset_converted_externally_to_rlds
austin_sailor_dataset_converted_externally_to_rlds
austin_sirius_dataset_converted_externally_to_rlds
bc_z
berkeley_autolab_ur5
berkeley_cable_routing
berkeley_fanuc_manipulation
berkeley_gnm_cory_hall
berkeley_gnm_recon
berkeley_gnm_sac_son
berkeley_mvp_converted_externally_to_rlds
berkeley_rpt_converted_externally_to_rlds
bridge
bridge_data_msr
cmu_franka_exploration_dataset_converted_externally_to_rlds
cmu_play_fusion
cmu_stretch
columbia_cairlab_pusht_real
conq_hose_manipulation
dlr_edan_shared_control_converted_externally_to_rlds
dlr_sara_grid_clamp_converted_externally_to_rlds
dlr_sara_pour_converted_externally_to_rlds
dobbe
eth_agent_affordances
fmb
fractal20220817_data
iamlab_cmu_pickup_insert_converted_externally_to_rlds
imperialcollege_sawyer_wrist_cam
io_ai_tech
jaco_play
kaist_nonprehensile_converted_externally_to_rlds
kuka
maniskill_dataset_converted_externally_to_rlds
mimic_play
mt_opt
nyu_door_opening_surprising_effectiveness
nyu_franka_play_dataset_converted_externally_to_rlds
nyu_rot_dataset_converted_externally_to_rlds
plex_robosuite
robo_set
roboturk
spoc_robot
stanford_hydra_dataset_converted_externally_to_rlds
stanford_kuka_multimodal_dataset_converted_externally_to_rlds
stanford_mask_vit_converted_externally_to_rlds
stanford_robocook_converted_externally_to_rlds
taco_play
tidybot
tokyo_u_lsmo_converted_externally_to_rlds
toto
ucsd_kitchen_dataset_converted_externally_to_rlds
ucsd_pick_and_place_dataset_converted_externally_to_rlds
uiuc_d3field
usc_cloth_sim_converted_externally_to_rlds
utaustin_mutex
utokyo_pr2_opening_fridge_converted_externally_to_rlds
utokyo_pr2_tabletop_manipulation_converted_externally_to_rlds
utokyo_saytap_converted_externally_to_rlds
utokyo_xarm_bimanual_converted_externally_to_rlds
utokyo_xarm_pick_and_place_converted_externally_to_rlds
vima_converted_externally_to_rlds
viola
Scene classification
Semantic segmentation
bigearthnet
cityscapes
kitti
lost_and_found
nyu_depth_v2
open_images_v4
places365_small
ref_coco
scene_parse150
segment_anything
so2sat
Sentiment analysis
Sequence modeling
Sequence to sequence language modeling
Speech
Speech recognition
Structured
cherry_blossoms
covid19
cs_restaurants
dart
diamonds
forest_fires
genomics_ood
german_credit_numeric
higgs
howell
iris
movie_lens
movielens
web_graph
web_nlg
wiki_bio
wiki_table_questions
wiki_table_text
wine_quality
Summarization
Table to text generation
Tabular
Text
abstract_reasoning
aeslc
ag_news_subset
ai2_arc
ai2_arc_with_ir
amazon_us_reviews
anli
answer_equivalence
arc
asqa
asset
assin2
bccd
beir
big_patent
billsum
blimp
booksum
bool_q
bot_adversarial_dialogue
bucc
c4
c4_wsrs
caltech101
cfq
cityscapes
civil_comments
clevr
clinc_oos
conll2002
conll2003
corr2cause
cos_e
databricks_dolly
definite_pronoun_resolution
dices
doc_nli
dolma
dolphin_number_word
drop
dsprites
e2e_cleaned
eraser_multi_rc
esnli
flic
gap
gem
glue
goemotions
gpt3
gsm8k
hellaswag
imdb_reviews
irc_disentanglement
kitti
lambada
lfw
librispeech
librispeech_lm
libritts
ljspeech
lm1b
lost_and_found
math_dataset
math_qa
mctaco
media_sum
mlqa
movie_rationales
mrqa
multi_nli
multi_nli_mismatch
natural_instructions
natural_questions
natural_questions_open
newsroom
open_images_challenge2019_detection
open_images_v4
openbookqa
opinion_abstracts
opinosis
opus
oxford_flowers102
oxford_iiit_pet
para_crawl
patch_camelyon
paws_wiki
paws_x_wiki
penguins
pet_finder
pg19
piqa
places365_small
placesfull
plant_leaves
plant_village
plantae_k
protein_net
q_re_cc
qa4mre
qasc
quac
quality
race
real_toxicity_prompts
reddit_disentanglement
reddit_tifu
ref_coco
resisc45
robonet
rock_you
salient_span_wikipedia
samsum
scan
schema_guided_dialogue
sci_tail
scicite
scientific_papers
scrolls
sentiment140
smallnorb
snli
spoken_digit
squad
squad_question_generation
stanford_dogs
star_cfq
story_cloze
summscreen
sun397
super_glue
svhn_cropped
tatoeba
ted_hrlr_translate
ted_multi_translate
tedlium
tiny_shakespeare
trec
trivia_qa
unified_qa
universal_dependencies
unnatural_instructions
user_libri_text
webvid
wiki40b
wiki_dialog
wikiann
wikipedia
wikipedia_toxicity_subtypes
winogrande
wordnet
wsc273
xnli
xtreme_pawsx
xtreme_xnli
yelp_polarity_reviews
Text classification
ag_news_subset
bool_q
bot_adversarial_dialogue
dices
imdb_reviews
natural_instructions
paws_wiki
paws_x_wiki
sentiment140
trec
Text classification toxicity prediction
Text generation
aeslc
big_patent
billsum
booksum
bool_q
databricks_dolly
e2e_cleaned
lambada
lm1b
math_qa
mctaco
natural_instructions
natural_questions
newsroom
openbookqa
piqa
race
real_toxicity_prompts
reddit
reddit_tifu
samsum
schema_guided_dialogue
scientific_papers
squad
story_cloze
trivia_qa
Text simplification
Text summarization
aeslc
big_patent
billsum
booksum
databricks_dolly
newsroom
reddit
reddit_tifu
samsum
scientific_papers
Time series
Token classification
Tracking
Trajectory
Translate
flores
mtnt
wmt13_translate
wmt14_translate
wmt15_translate
wmt16_translate
wmt17_translate
wmt18_translate
wmt19_translate
wmt_t2t_translate
Uncategorized
Unsupervised anomaly detection
Video
abstract_reasoning
bair_robot_pushing_small
davis
flic
moving_mnist
robonet
starcraft_video
tao
ucf101
webvid
youtube_vis