noether.data.datasets.cfd.caeml

Submodules

Attributes

Classes

AhmedMLDataset

Dataset implementation for AhmedML CFD simulations.

AhmedMLDefaultSplitIDs

Default split IDs for AhmedML dataset with validation.

DrivAerMLDataset

Dataset implementation for DrivaerML CFD simulations.

DrivAerMLDefaultSplitIDs

Default split IDs for DrivAerML dataset with validation.

Package Contents

class noether.data.datasets.cfd.caeml.AhmedMLDataset(dataset_config)

Bases: noether.data.datasets.cfd.caeml.dataset.CAEMLDataset

Dataset implementation for AhmedML CFD simulations.

Parameters:

dataset_config (noether.core.schemas.dataset.DatasetBaseConfig) – Configuration for the dataset.

Initialize the AhmedML dataset.

Parameters:

dataset_config (noether.core.schemas.dataset.DatasetBaseConfig) – Configuration for the dataset.

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

noether.core.schemas.dataset.DatasetSplitIDs

class noether.data.datasets.cfd.caeml.AhmedMLDefaultSplitIDs(/, **data)

Bases: noether.core.schemas.dataset.DatasetSplitIDs

Default split IDs for AhmedML dataset with validation.

Create a new model by parsing and validating input data from keyword arguments.

Raises [ValidationError][pydantic_core.ValidationError] if the input data cannot be validated to form a valid model.

self is explicitly positional-only to allow self as a field name.

Parameters:

data (Any)

EXPECTED_TRAIN_SIZE = 400
EXPECTED_VAL_SIZE = 50
EXPECTED_TEST_SIZE = 50
DATASET_NAME = 'AhmedML'
train: set[int]
val: set[int]
test: set[int]
static create_split()

A helper function to create a random split of the dataset. The default indices were created with seed=42.

class noether.data.datasets.cfd.caeml.DrivAerMLDataset(dataset_config)

Bases: noether.data.datasets.cfd.caeml.dataset.CAEMLDataset

Dataset implementation for DrivaerML CFD simulations.

Parameters:

dataset_config (noether.core.schemas.dataset.DatasetBaseConfig) – Configuration for the dataset.

Initialize the DrivaerML dataset.

Parameters:

dataset_config (noether.core.schemas.dataset.DatasetBaseConfig) – Configuration for the dataset.

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

noether.core.schemas.dataset.DatasetSplitIDs

class noether.data.datasets.cfd.caeml.DrivAerMLDefaultSplitIDs(/, **data)

Bases: noether.core.schemas.dataset.DatasetSplitIDs

Default split IDs for DrivAerML dataset with validation.

Create a new model by parsing and validating input data from keyword arguments.

Raises [ValidationError][pydantic_core.ValidationError] if the input data cannot be validated to form a valid model.

self is explicitly positional-only to allow self as a field name.

Parameters:

data (Any)

EXPECTED_TRAIN_SIZE = 400
EXPECTED_VAL_SIZE = 34
EXPECTED_TEST_SIZE = 50
EXPECTED_HIDDEN_TEST_SIZE = 16
DATASET_NAME = 'DrivAerML'
train: set[int]
val: set[int]
test: set[int]
hidden_test: set[int]
static create_split()

A helper function to create a random split of the dataset. The default indices were created with seed=42.

noether.data.datasets.cfd.caeml.CAEML_FILEMAP