guess_the_correlation_dataset: Guess The Correlation dataset

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

Guess The Correlation dataset

Description

Prepares the Guess The Correlation dataset available on Kaggle here A copy of this dataset is hosted in a public Google Cloud bucket so you don't need to authenticate.

Usage

guess_the_correlation_dataset(
  root,
  split = "train",
  transform = NULL,
  target_transform = NULL,
  indexes = NULL,
  download = FALSE
)

Arguments

root	path to the data location
split	string. 'train' or 'submission'
transform	function that takes a torch tensor representing an image and return another tensor, transformed.
target_transform	function that takes a scalar torch tensor and returns another tensor, transformed.
indexes	set of integers for subsampling (e.g. 1:140000)
download	whether to download or not

Value

A torch dataset that can be consumed with torch::dataloader().

Examples

if (torch::torch_is_installed() && FALSE) {
gtc <- guess_the_correlation_dataset("./data", download = TRUE)
length(gtc)
}

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