noether.data.datasets.cfd.caeml.ahmedml

Submodules

• noether.data.datasets.cfd.caeml.ahmedml.dataset
• noether.data.datasets.cfd.caeml.ahmedml.split

Classes

AhmedMLDataset	Dataset implementation for AhmedML CFD simulations.
AhmedMLDefaultSplitIDs	Default split IDs for AhmedML dataset with validation.

Package Contents

class noether.data.datasets.cfd.caeml.ahmedml.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.ahmedml.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.