torchdatasets

<!-- badges: start -->

<!-- badges: end -->

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 | | spam_dataset() | Tabular | Classification | Not required |