acds.benchmarks.adiac package

Submodules

acds.benchmarks.adiac.dataset module

class acds.benchmarks.adiac.dataset.AdiacDataset(data: Tuple[Tensor, Tensor])[source]

Bases: Dataset

Dataset wrapper for Adiac time-series samples.

This class assumes mydata to have the form [(x1, y1), (x2, y2)], where xi are inputs, and yi are targets.

acds.benchmarks.adiac.getter module

acds.benchmarks.adiac.getter.get_adiac_data(root_path: PathLike, bs_train: int, bs_test: int, whole_train: bool = False, for_rc: bool = True) → tuple[DataLoader, DataLoader, DataLoader][source]

Get the ADIAC dataset from a txt file and return the train, validation and test dataloaders.

Parameters:

  • root_path (os.PathLike) – Path to the folder containing the txt files.

  • bs_train (int) – Batch size for the train dataloader.

  • bs_test (int) – Batch size for the validation and test dataloaders.

  • whole_train (bool, optional) – If True, the whole dataset is used for training. Defaults to False.

  • for_rc (bool, optional) – If True, the data is returned as a RCDataset. Defaults to True.

Returns:

Train, validation and test dataloaders.

Return type:

Tuple[DataLoader, DataLoader, DataLoader]

Module contents