neuralop.data.datasets.navier_stokes.NavierStokesDataset

class neuralop.data.datasets.navier_stokes.NavierStokesDataset(root_dir: Path | str, n_train: int, n_tests: List[int], batch_size: int, test_batch_sizes: List[int], train_resolution: int, test_resolutions: int = [16, 32], encode_input: bool = False, encode_output: bool = True, encoding='channel-wise', channel_dim=1, subsampling_rate=None, download: bool = True)[source]

NavierStokesDataset stores data generated according to the 2d incompressible Navier-Stokes equations. Input and output are both 2d fields with one channel of data which describes the vorticity at each point.

Data source: https://zenodo.org/records/12825163

Attributes:

train_db: torch.utils.data.Dataset of training examples

test_db: “” of test examples

data_processor: neuralop.data.transforms.DataProcessor to process data examples

optional, default is None