torchdatasets provides ready-to-use datasets compatible with the torch package.
The released version of torchdatasets can be installed with:
install.packages("torchdatasets")
You can also install the development version with:
# remotes::install_github("mlverse/torchdatasets")
install.packages("torchdatasets", repos = c('https://mlverse.r-universe.dev', 'https://cloud.r-project.org'))
Currently, the following datasets are implemented:
| Dataset | Domain | Type | Authentication |
|---|---|---|---|
| bird_species_dataset() | Images | Classification | Not required |
| dogs_vs_cats_dataset() | Images | Classification | Not required |
| guess_the_correlation_dataset() | Images | Regression | Not required |
| cityscapes_pix2pix_dataset() | Images | Segmentation | Not required |
| oxford_pet_dataset() | Images | Segmentation | Not required |
| bank_marketing_dataset() | Tabular | Classification | Not required |
| imdb_dataset() | Text | Classification | Not required |
| spam_dataset() | Tabular | Classification | Not required |