https://github.com/mlverse/torchdatasets

Extra datasets for torch
https://github.com/mlverse/torchdatasets

Last synced: 12 months ago
JSON representation

Extra datasets for torch

• Host: GitHub
• URL: https://github.com/mlverse/torchdatasets
• Owner: mlverse
• License: other
• Created: 2020-09-29T19:26:47.000Z (almost 6 years ago)
• Default Branch: main
• Last Pushed: 2025-03-11T14:21:03.000Z (over 1 year ago)
• Last Synced: 2025-07-07T13:40:41.136Z (12 months ago)
• Language: R
• Homepage: https://mlverse.github.io/torchdatasets
• Size: 343 KB
• Stars: 16
• Watchers: 3
• Forks: 6
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

          
# torchdatasets

[![R-CMD-check](https://github.com/mlverse/torchdatasets/workflows/R-CMD-check/badge.svg)](https://github.com/mlverse/torchdatasets/actions) [![CRAN status](https://www.r-pkg.org/badges/version/torchdatasets)](https://CRAN.R-project.org/package=torchdatasets) [![](https://cranlogs.r-pkg.org/badges/torchdatasets)](https://cran.r-project.org/package=torchdatasets)

torchdatasets provides ready-to-use datasets compatible with the [torch](https://github.com/mlverse/torch) package.

## Installation

The released version of torchdatasets can be installed with:

``` r

install.packages("torchdatasets")

```

You can also install the development version with:

``` r

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   |