README.md

In dfalbel/extradatasets: Ready to Use Extra Datasets for Torch

torchdatasets

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 |

dfalbel/extradatasets documentation built on April 12, 2025, 9:36 a.m.