deepsphere.data.datasets package

Submodules

deepsphere.data.datasets.dataset module

Datasets 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 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

property 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'

class deepsphere.data.datasets.dataset.ARTCTemporaldataset(path, sequence_length, prediction_shift=0, indices=None, transform_image=None, transform_labels=None, transform_sample=None, download=False)

Bases: deepsphere.data.datasets.dataset.ARTCDataset

Dataset for reduced ARTC dataset with temporality functionality.

Module contents