The WiML Symposium 2023 returns December 7! Join the livestream at 9:30 AM PST Register

Datasets

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`

`Biology`

ogbg_molpcba

`Categorical`

sift1m

`Common sense reasoning`

`Computer science`

`Conditional image generation`

`Coreference resolution`

clevr

`D4rl`

`Density estimation`

`Dependency parsing`

`Dialog act labeling`

bot_adversarial_dialogue

`Dialogue`

`Document summarization`

`Fine grained image classification`

`Graph`

`Graphs`

cardiotox

`Image`

`Image classification`

`Image clustering`

`Image compression`

`Image generation`

`Image segmentation`

segment_anything

`Image super resolution`

`Image to image translation`

`Instance segmentation`

`Language modeling`

`Linguistic acceptability`

bot_adversarial_dialogue

`Machine translation`

`Monolingual`

`Movies and tv shows`

opinion_abstracts

`Multilingual`

`Natural language inference`

`Natural language understanding`

`Nearest neighbors`

`Object detection`

`Open domain question answering`

`Out of distribution detection`

stl10

`Question answering`

`Question generation`

`Ranking`

`Reading comprehension`

`Recommendation`

`Reinforcement learning`

`Rgb d`

nyu_depth_v2

`Rl unplugged`

`Rlds`

`Robotics`

`Scene classification`

`Semantic segmentation`

`Sentiment analysis`

`Sequence modeling`

databricks_dolly

`Sequence to sequence language modeling`

`Speech`

`Speech recognition`

`Structured`

`Summarization`

`Table to text generation`

e2e_cleaned

`Tabular`

`Text`

`Text classification`

`Text classification toxicity prediction`

`Text generation`

`Text simplification`

wiki_auto

`Text summarization`

`Time series`

`Token classification`

`Tracking`

smartwatch_gestures

`Trajectory`

`Translate`

`Uncategorized`

duke_ultrasound

`Unsupervised anomaly detection`

caltech101

`Video`

`Vision language`