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 Kaggle (via pins)
dogs_vs_cats_dataset() Images Classification Kaggle (via pins)
guess_the_correlation_dataset() Images Regression Not required