noether.data.datasets.cfd.emmi_wing¶

Submodules¶

Classes¶

EmmiWingDataset

Dataset implementation for aerodynamic datasets with volume and surface fields.

Package Contents¶

class noether.data.datasets.cfd.emmi_wing.EmmiWingDataset(dataset_config)¶

Bases: noether.data.datasets.cfd.dataset.AeroDataset

Dataset implementation for aerodynamic datasets with volume and surface fields. This unified dataset class provides an interface for aerodynamics dataset with volume and surface fields. The dataset behavior such as the dataset choice, train/val/test split IDs, etc. is configured through constructor parameters, allowing for easy extension to new datasets.

Parameters:

dataset_config (noether.core.schemas.dataset.StandardDatasetConfig) – Configuration for the dataset. See DatasetBaseConfig for available options.
filemap – FileMap object defining the mapping of data properties to filenames. See FileMap for details.

split¶

source_root¶

sample_info(idx)¶

Get information about a sample such as its local path, run name, etc.

Parameters:: idx (int)
Return type:: dict[str, str | int | None]

property get_dataset_splits: noether.core.schemas.dataset.DatasetSplitIDs¶

Return type:: noether.core.schemas.dataset.DatasetSplitIDs

property supported_splits: set[str]¶

Return type:: set[str]

getitem_geometry_design_parameters(idx)¶

Retrieves geometry design parameters as a single tensor.

Returns:: Geometry design parameters tensor of shape (1, num_geometry_parameters)
Return type:: torch.Tensor
Parameters:: idx (int)

getitem_inflow_design_parameters(idx)¶

Retrieves inflow design parameters as a single tensor.

Returns:: Inflow design parameters tensor of shape (1, num_inflow_parameters)
Return type:: torch.Tensor
Parameters:: idx (int)