neuralop.data.datasets.burgers.load_mini_burgers_1dtime
- neuralop.data.datasets.burgers.load_mini_burgers_1dtime(data_path: Path | str, n_train: int, n_test: int, batch_size: int, test_batch_size: int, temporal_subsample: int = 1, spatial_subsample: int = 1)[source]
Legacy function to load mini Burger’s equation dataset
- Parameters:
- root_dirUnion[Path, str]
root at which to download data files
- n_trainint
number of train instances
- n_testint
number of test instances per test dataset
- batch_sizeint
batch size of training set
- test_batch_sizeint
batch size of test set
- temporal_subsampleint, optional
rate at which to subsample the temporal dimension, by default None
- spatial_subsampleint, optional
rate at which to subsample along the spatial dimension, by default None