torchdatasets

R-CMD-check CRAN status

torchdatasets provides ready-to-use datasets compatible with the torch package.

Installation

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")

Datasets

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