使用法
簡単な紹介については、スタートガイドをご覧ください。
for ex in tfds.load('cifar10', split='train'):
...
すべてのデータセット
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 -
ai2dcaption -
bccd -
beans -
bee_dataset -
bigearthnet -
binarized_mnist -
binary_alpha_digits -
caltech101 -
celeb_a -
celeb_a_hq -
cityscapes -
clevr -
clic -
coil100 -
covr -
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 -
imagenet2012_subset -
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 -
imagenet2012_subset -
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 -
covr -
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 -
covr -
databricks_dolly -
dices -
imdb_reviews -
math_dataset -
math_qa -
mlqa -
multi_news -
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
News
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 -
asimov_dilemmas_auto_val -
asimov_dilemmas_scifi_train -
asimov_dilemmas_scifi_val -
asimov_injury_val -
asimov_multimodal_auto_val -
asimov_multimodal_manual_val -
asimov_v2_constraints_with_rationale -
asimov_v2_constraints_without_rationale -
asimov_v2_injuries -
asimov_v2_videos -
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 -
covr -
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_news -
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 -
multi_news -
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