Index

A

accumulate_metrics() (training.adapters.GraphAdapter method) (training.adapters.GridAdapter method) (training.adapters.ModelAdapter method) (training.adapters.PointwiseAdapter method) (training.adapters.ProfileAdapter method) adapter (training.models.ModelEntry attribute) add_baseline_to_encoded() (cases.alpha_d.datasets.profile.AlphaDProfileDataset method) (training.datasets_tabular.TabularPairDataset method) alpha_d_baseline_profile() (in module cases.alpha_d.physics.baseline)

alpha_d_bulk_to_values() (in module cases.alpha_d.physics.targets) alpha_d_residual_transform() (in module cases.alpha_d.transforms) alpha_d_values_to_bulk() (in module cases.alpha_d.physics.targets) AlphaDConv1D (class in training.models.conv1d_profile) AlphaDExperiment (class in cases.alpha_d.experiment) AlphaDProfileDataset (class in cases.alpha_d.datasets.profile) AlphaDSource (class in cases.alpha_d.etl.source) AlphaDTransformation (class in cases.alpha_d.etl.transform) AlphaDZarrSink (class in cases.alpha_d.etl.sink) apply_overrides() (in module training.hpo.search_space)

B

baseline_for_plotting() (cases.alpha_d.experiment.AlphaDExperiment method) (training.experiment.Experiment method) BaselineGeometry (class in cases.alpha_d.physics.baseline) buffer_diams (cases.alpha_d.physics.baseline.BaselineGeometry attribute) build() (in module training.models.afno) (in module training.models.conv1d_profile) (in module training.models.fno) (in module training.models.meshgraphnet) (in module training.models.mlp) (in module training.models.pix2pix) build_batch() (training.adapters.GraphAdapter method) (training.adapters.GridAdapter method) (training.adapters.ModelAdapter method) (training.adapters.PointwiseAdapter method) (training.adapters.ProfileAdapter method)

build_dataframe() (in module feature_selection.pycaret_selection) build_dataset() (in module cases.alpha_d.datasets.profile) (training.adapters.GraphAdapter method) (training.adapters.GridAdapter method) (training.adapters.ModelAdapter method) (training.adapters.PointwiseAdapter method) (training.adapters.ProfileAdapter method) build_engineered_feature_map() (in module cases.alpha_d.feature_data) build_experiment() (in module training.runner) build_fn (training.models.ModelEntry attribute) build_manifest() (in module feature_selection.manifest)

C

case_ids (feature_selection.data.FeatureAnalysisData attribute) case_level_split() (in module feature_selection.pycaret_selection) cases.alpha_d.datasets.profile module cases.alpha_d.etl.sink module cases.alpha_d.etl.source module cases.alpha_d.etl.transform module cases.alpha_d.experiment module cases.alpha_d.feature_data module cases.alpha_d.metrics module cases.alpha_d.physics.baseline module cases.alpha_d.physics.targets module cases.alpha_d.transforms module cases.case_pressure_drop.feature_selection module cases.moose_grid.etl.data_sources.csv_source module

cases.moose_grid.etl.data_sources.exodus_source module cases.moose_grid.etl.data_sources.zarr_sink module cases.moose_grid.etl.schemas module cases.moose_grid.etl.transformations.moose_transform module cases.moose_grid.etl.validators module cleanup_temp_files() (cases.alpha_d.etl.sink.AlphaDZarrSink method) (cases.moose_grid.etl.data_sources.zarr_sink.MOOSEZarrSink method) collate_fn() (training.adapters.GraphAdapter method) (training.adapters.ModelAdapter method) compute_delta_p_metrics() (in module cases.alpha_d.metrics) compute_extended_metrics() (cases.alpha_d.experiment.AlphaDExperiment method) (training.experiment.Experiment method) compute_pointwise_extended_metrics() (in module cases.alpha_d.metrics) compute_val_loss() (in module training.runner) connectivity (cases.moose_grid.etl.schemas.MOOSEProcessedData attribute) (cases.moose_grid.etl.schemas.MOOSERawData attribute) Conv1DProfile (class in training.models.conv1d_profile) convert_alpha_d_values_between_bases() (in module cases.alpha_d.physics.targets) coords (cases.moose_grid.etl.schemas.MOOSEProcessedData attribute) (cases.moose_grid.etl.schemas.MOOSERawData attribute) create_study() (in module training.hpo.study) CSVProbeSource (class in cases.moose_grid.etl.data_sources.csv_source)

D

D_big (cases.alpha_d.physics.baseline.BaselineGeometry attribute) dataset.moose_dataset module dataset_info() (training.adapters.GraphAdapter method) (training.adapters.GridAdapter method) (training.adapters.ModelAdapter method) (training.adapters.PointwiseAdapter method) (training.adapters.ProfileAdapter method)

decode_alpha_d_target() (in module cases.alpha_d.physics.targets) decode_for_plotting() (cases.alpha_d.experiment.AlphaDExperiment method) (training.experiment.Experiment method) denormalize() (dataset.moose_dataset.MOOSEDataset method) Dr (cases.alpha_d.physics.baseline.BaselineGeometry attribute) dz_norm (cases.alpha_d.physics.baseline.BaselineGeometry property) dz_phys (cases.alpha_d.physics.baseline.BaselineGeometry property)

E

edge_dst (cases.moose_grid.etl.schemas.MOOSEProcessedData attribute) edge_src (cases.moose_grid.etl.schemas.MOOSEProcessedData attribute) elem_to_node() (in module dataset.moose_dataset) encode_alpha_d_target() (in module cases.alpha_d.physics.targets) enforce_allowlist() (in module feature_selection.pycaret_selection) engineered_features_spec() (in module cases.alpha_d.feature_data)

eval_step() (training.experiment.Experiment method) evaluate() (in module training.runner) exclude_cases (cases.alpha_d.datasets.profile.AlphaDProfileDataset property) ExodusDataSource (class in cases.moose_grid.etl.data_sources.exodus_source) Experiment (class in training.experiment) export_split_files() (in module training.split_io) export_split_files_from_run_meta() (in module training.split_io)

F

family (training.adapters.GraphAdapter attribute) (training.adapters.GridAdapter attribute) (training.adapters.ModelAdapter attribute) (training.adapters.PointwiseAdapter attribute) (training.adapters.ProfileAdapter attribute) feature_names (feature_selection.data.FeatureAnalysisData attribute) feature_selection.data module feature_selection.manifest module feature_selection.pycaret_selection module FeatureAnalysisData (class in feature_selection.data) field_names (cases.moose_grid.etl.schemas.MOOSEProcessedData attribute) (cases.moose_grid.etl.schemas.MOOSERawData attribute)

field_values_to_physical() (in module cases.alpha_d.physics.targets) fields (cases.moose_grid.etl.schemas.MOOSEProcessedData attribute) (cases.moose_grid.etl.schemas.MOOSERawData attribute) find_probe_files() (in module cases.moose_grid.etl.data_sources.csv_source) forward() (training.models.conv1d_profile.Conv1DProfile method) forward_eval() (training.adapters.GraphAdapter method) (training.adapters.GridAdapter method) (training.adapters.ModelAdapter method) (training.adapters.PointwiseAdapter method) (training.adapters.ProfileAdapter method) forward_train() (training.adapters.GraphAdapter method) (training.adapters.GridAdapter method) (training.adapters.ModelAdapter method) (training.adapters.PointwiseAdapter method) (training.adapters.ProfileAdapter method)

G

get_adapter() (in module training.adapters) get_build_fn_and_adapter() (in module training.models) get_file_list() (cases.alpha_d.etl.sink.AlphaDZarrSink method) (cases.alpha_d.etl.source.AlphaDSource method) (cases.moose_grid.etl.data_sources.exodus_source.ExodusDataSource method) (cases.moose_grid.etl.data_sources.zarr_sink.MOOSEZarrSink method) get_loss_fn() (in module training.losses)

GraphAdapter (class in training.adapters) GraphPairDataset (class in training.datasets) grid_fields (cases.moose_grid.etl.schemas.MOOSEProcessedData attribute) grid_x (cases.moose_grid.etl.schemas.MOOSEProcessedData attribute) grid_y (cases.moose_grid.etl.schemas.MOOSEProcessedData attribute) GridAdapter (class in training.adapters) GridPairDataset (class in training.datasets) groups (feature_selection.data.FeatureAnalysisData attribute)

H

has_target_baseline (cases.alpha_d.datasets.profile.AlphaDProfileDataset property)

I

import_physicsnemo_attr() (in module training) import_physicsnemo_module() (in module training) in_features (cases.alpha_d.datasets.profile.AlphaDProfileDataset property) (training.datasets_tabular.TabularPairDataset property)

input_columns (cases.alpha_d.datasets.profile.AlphaDProfileDataset property) integrated_baseline_delta_p() (in module cases.alpha_d.physics.baseline) is_alpha_d_target() (in module cases.alpha_d.physics.targets)

L

l1_loss() (in module training.losses) L_roi (cases.alpha_d.physics.baseline.BaselineGeometry property) load_feature_matrix() (in module cases.alpha_d.feature_data) load_fields() (in module dataset.moose_dataset)

load_norm_stats() (in module dataset.moose_dataset) load_run_meta() (in module training.split_io) local_velocity_normalization (cases.alpha_d.datasets.profile.AlphaDProfileDataset property) (feature_selection.data.FeatureAnalysisData attribute) Lr (cases.alpha_d.physics.baseline.BaselineGeometry attribute)

M

make_objective() (in module training.hpo.objective) manifest_path (cases.case_pressure_drop.feature_selection.SelectionResult attribute) mean (cases.moose_grid.etl.schemas.NormStats attribute) mean_rank (cases.case_pressure_drop.feature_selection.MethodResult attribute) mean_score (cases.case_pressure_drop.feature_selection.MethodResult attribute) MethodResult (class in cases.case_pressure_drop.feature_selection) model_entrypoint_string() (in module training.models) ModelAdapter (class in training.adapters) ModelEntry (class in training.models) MODES (dataset.moose_dataset.MOOSEDataset attribute) module cases.alpha_d.datasets.profile cases.alpha_d.etl.sink cases.alpha_d.etl.source cases.alpha_d.etl.transform cases.alpha_d.experiment cases.alpha_d.feature_data cases.alpha_d.metrics cases.alpha_d.physics.baseline cases.alpha_d.physics.targets cases.alpha_d.transforms cases.case_pressure_drop.feature_selection cases.moose_grid.etl.data_sources.csv_source cases.moose_grid.etl.data_sources.exodus_source cases.moose_grid.etl.data_sources.zarr_sink cases.moose_grid.etl.schemas cases.moose_grid.etl.transformations.moose_transform cases.moose_grid.etl.validators dataset.moose_dataset feature_selection.data feature_selection.manifest feature_selection.pycaret_selection training training.adapters training.datasets training.datasets_tabular training.experiment training.hpo training.hpo.objective training.hpo.search_space training.hpo.study training.hpo.visualize training.losses training.models training.models.afno training.models.conv1d_profile training.models.fno training.models.meshgraphnet training.models.mlp training.models.pix2pix training.plotting training.runner training.split_io

MOOSEDataset (class in dataset.moose_dataset) MOOSEDatasetValidator (class in cases.moose_grid.etl.validators) MOOSEDataTransformation (class in cases.moose_grid.etl.transformations.moose_transform) MOOSEProcessedData (class in cases.moose_grid.etl.schemas) MOOSERawData (class in cases.moose_grid.etl.schemas) MOOSEZarrSink (class in cases.moose_grid.etl.data_sources.zarr_sink) mse_loss() (in module training.losses)

N

n_cases (feature_selection.data.FeatureAnalysisData property) n_stations (cases.alpha_d.physics.baseline.BaselineGeometry attribute) norm_stats (cases.alpha_d.datasets.profile.AlphaDProfileDataset property) (cases.moose_grid.etl.schemas.MOOSEProcessedData attribute)

normalize (cases.alpha_d.datasets.profile.AlphaDProfileDataset property) normalize_split_cfg() (in module training.runner) NormStats (class in cases.moose_grid.etl.schemas)

O

on_epoch_end() (training.experiment.Experiment method) on_epoch_end_extra_step() (training.experiment.Experiment method) out_features (cases.alpha_d.datasets.profile.AlphaDProfileDataset property) (training.datasets_tabular.TabularPairDataset property)

outer_height_m (cases.alpha_d.physics.baseline.BaselineGeometry attribute) output_columns (cases.alpha_d.datasets.profile.AlphaDProfileDataset property)

P

parse_field_list() (in module training.datasets) parse_index_list() (in module training.plotting) per_fold_ranks (cases.case_pressure_drop.feature_selection.MethodResult attribute) PointwiseAdapter (class in training.adapters) prepare_for_training() (cases.alpha_d.experiment.AlphaDExperiment method) (training.experiment.Experiment method) prepare_training() (in module training.runner)

print_extended_metrics() (cases.alpha_d.experiment.AlphaDExperiment method) (in module cases.alpha_d.metrics) (training.experiment.Experiment method) probe_columns (cases.moose_grid.etl.schemas.MOOSEProcessedData attribute) (cases.moose_grid.etl.schemas.MOOSERawData attribute) probe_data (cases.moose_grid.etl.schemas.MOOSEProcessedData attribute) (cases.moose_grid.etl.schemas.MOOSERawData attribute) ProfileAdapter (class in training.adapters)

R

Re (cases.alpha_d.physics.baseline.BaselineGeometry attribute) read_all() (cases.moose_grid.etl.data_sources.csv_source.CSVProbeSource method) read_csv() (in module cases.moose_grid.etl.data_sources.csv_source) read_file() (cases.alpha_d.etl.sink.AlphaDZarrSink method) (cases.alpha_d.etl.source.AlphaDSource method) (cases.moose_grid.etl.data_sources.exodus_source.ExodusDataSource method) (cases.moose_grid.etl.data_sources.zarr_sink.MOOSEZarrSink method) register_model() (in module training.models) relative_l2_loss() (in module training.losses) report (cases.case_pressure_drop.feature_selection.SelectionResult attribute)

report_path (cases.case_pressure_drop.feature_selection.SelectionResult attribute) resolve_device() (in module training.runner) resolve_entrypoint() (in module training.models) resolve_plot_indices() (in module training.plotting) resolve_time_idx() (in module training.datasets) rho (cases.alpha_d.physics.baseline.BaselineGeometry attribute) rows_per_case (feature_selection.data.FeatureAnalysisData attribute) run_feature_selection() (in module cases.case_pressure_drop.feature_selection) run_hpo() (in module training.hpo.study) run_pycaret_selection() (in module feature_selection.pycaret_selection)

S

sample_from_search_space() (in module training.hpo.search_space) save_delta_p_parity_plot() (in module training.plotting) save_grid_prediction_plots() (in module training.plotting) save_parity_plot() (in module training.plotting) save_pointwise_profile_plots() (in module training.plotting) save_profile_prediction_plots() (in module training.plotting) save_study_plots() (in module training.hpo.visualize) select_best_worst_pointwise_cases() (in module training.plotting) selected_features (cases.case_pressure_drop.feature_selection.SelectionResult attribute) selected_features_path (cases.case_pressure_drop.feature_selection.SelectionResult attribute) SelectionResult (class in cases.case_pressure_drop.feature_selection) set_seed() (in module training.runner)

should_skip() (cases.alpha_d.etl.sink.AlphaDZarrSink method) (cases.moose_grid.etl.data_sources.zarr_sink.MOOSEZarrSink method) sim_name (cases.moose_grid.etl.schemas.MOOSEProcessedData attribute) (cases.moose_grid.etl.schemas.MOOSERawData attribute) sim_names (cases.alpha_d.datasets.profile.AlphaDProfileDataset property) (training.datasets_tabular.TabularPairDataset property) slice_time() (in module dataset.moose_dataset) split_indices() (in module training.datasets) stability (cases.case_pressure_drop.feature_selection.MethodResult attribute) std (cases.moose_grid.etl.schemas.NormStats attribute) subset_by_case_indices() (cases.alpha_d.datasets.profile.AlphaDProfileDataset method) (training.datasets_tabular.TabularPairDataset method)

T

TabularPairDataset (class in training.datasets_tabular) target_name (feature_selection.data.FeatureAnalysisData attribute) time_steps (cases.moose_grid.etl.schemas.MOOSEProcessedData attribute) (cases.moose_grid.etl.schemas.MOOSERawData attribute) to_plain_dict() (in module training.runner) to_tensor() (in module dataset.moose_dataset) train() (in module training.runner) train_one_epoch() (in module training.runner) train_or_hpo() (in module training.runner) training module training.adapters module training.datasets module training.datasets_tabular module training.experiment module training.hpo module training.hpo.objective module training.hpo.search_space module training.hpo.study module

training.hpo.visualize module training.losses module training.models module training.models.afno module training.models.conv1d_profile module training.models.fno module training.models.meshgraphnet module training.models.mlp module training.models.pix2pix module training.plotting module training.runner module training.split_io module training_step() (training.experiment.Experiment method) transform() (cases.alpha_d.etl.transform.AlphaDTransformation method) (cases.moose_grid.etl.transformations.moose_transform.MOOSEDataTransformation method)

U

used_case_ids (cases.case_pressure_drop.feature_selection.SelectionResult attribute)

V

V_bulk (cases.alpha_d.physics.baseline.BaselineGeometry attribute) validate() (cases.moose_grid.etl.validators.MOOSEDatasetValidator method) validate_search_space() (in module training.hpo.search_space)

validate_single_item() (cases.moose_grid.etl.validators.MOOSEDatasetValidator method) validation_epoch_loss() (training.experiment.Experiment method) validation_step() (training.experiment.Experiment method)

W

write_manifest() (in module feature_selection.manifest)

write_selected_features() (in module feature_selection.pycaret_selection) write_sim_name_list() (in module training.split_io)

X

X (feature_selection.data.FeatureAnalysisData attribute)

Y

y (feature_selection.data.FeatureAnalysisData attribute)

Z

z_throat_end_norm (cases.alpha_d.physics.baseline.BaselineGeometry property)

z_throat_start_norm (cases.alpha_d.physics.baseline.BaselineGeometry property)