deepsphere.data.datasets package

Submodules

deepsphere.data.datasets.dataset module

Dataset for reduced atmospheric river and tropical cyclone detection dataset.

class deepsphere.data.datasets.dataset.ARTCDataset(path, indices=None, transform_data=None, transform_labels=None, download=False)

Bases: torch.utils.data.dataset.Dataset

Dataset for reduced atmospheric river and tropical cyclone dataset.

check_exists()

Check if dataset already exists.

download()

Download the dataset if it doesn’t already exist.

get_runs(runs)

Get datapoints corresponding to specific runs.

Parameters:runs (list) – List of desired runs. Returns:List of strings, which represents the files in the dataset, which belong to one of the desired runs. Return type:list

indices

Get files.

Returns:List of strings, which represent the files contained in the dataset. Return type:list

resource = 'http://island.me.berkeley.edu/ugscnn/data/climate_sphere_l5.zip'

Module contents