src.FRAME_FM.dataloaders.xarray_dataloader

Classes

TabularDataset	Very simple tabular dataset: X numeric features, y target.
GriddedDataModule	Loads regularly gridded datasets, inherits from the FRAME-FM base class BaseDataModule

Functions

main()

Module Contents

class src.FRAME_FM.dataloaders.xarray_dataloader.TabularDataset(*tensors: torch.Tensor)

Bases: torch.utils.data.TensorDataset

Very simple tabular dataset: X numeric features, y target.

class src.FRAME_FM.dataloaders.xarray_dataloader.GriddedDataModule(data_root: str, filename: str, feature_cols: list[str], target_col: str, val_fraction: float = 0.1, test_fraction: float = 0.1, **base_kwargs: Any)

Bases: FRAME_FM.utils.LightningDataModuleWrapper.BaseDataModule

Loads regularly gridded datasets, inherits from the FRAME-FM base class BaseDataModule

Expects: - A file at data_root / filename - Columns split into features / target via config

filename

feature_cols

target_col

val_fraction = 0.1

test_fraction = 0.1

src.FRAME_FM.dataloaders.xarray_dataloader.main()