Index

A

AbstractSurrogateBaseConfig (class in codes.surrogates.AbstractSurrogate), [1] (class in codes.surrogates.AbstractSurrogate.abstract_config), [1] AbstractSurrogateModel (class in codes.surrogates), [1] (class in codes.surrogates.AbstractSurrogate), [1] (class in codes.surrogates.AbstractSurrogate.abstract_surrogate), [1] acquire() (codes.train.DummyLock method), [1] (codes.train.train_fcts.DummyLock method), [1]

activation (codes.surrogates.AbstractSurrogate.abstract_config.AbstractSurrogateBaseConfig attribute), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateBaseConfig attribute), [1] apply_tuning_defaults() (in module codes.tune), [1] (in module codes.tune.tune_utils), [1] atol (codes.surrogates.LatentNeuralODE.latent_neural_ode_config.LatentNeuralODEBaseConfig attribute), [1]

B

batch_factor_to_float() (in module codes.utils), [1] (in module codes.utils.utils), [1] beta (codes.surrogates.AbstractSurrogate.abstract_config.AbstractSurrogateBaseConfig attribute), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateBaseConfig attribute), [1] branch_hidden_layers (codes.surrogates.DeepONet.deeponet_config.MultiONetBaseConfig attribute), [1]

BranchNet (class in codes.surrogates), [1] (class in codes.surrogates.DeepONet.deeponet), [1] build_fine_optuna_params() (in module codes.tune), [1] (in module codes.tune.optuna_fcts), [1] build_study_names() (in module codes.tune), [1] (in module codes.tune.tune_utils), [1]

C

check_and_load_data() (in module codes.utils), [1] (in module codes.utils.data_utils), [1] check_benchmark() (in module codes.benchmark), [1] (in module codes.benchmark.bench_utils), [1] check_surrogate() (in module codes.benchmark), [1] (in module codes.benchmark.bench_utils), [1] check_training_status() (in module codes.utils), [1] (in module codes.utils.utils), [1] checkpoint() (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel method), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel method), [1] (codes.surrogates.AbstractSurrogateModel method), [1] ChemDataset (class in codes.surrogates), [1] (class in codes.surrogates.LatentNeuralODE.utilities), [1] clean_metrics() (in module codes.benchmark), [1] (in module codes.benchmark.bench_utils), [1] coder_hidden (codes.surrogates.LatentPolynomial.latent_poly_config.LatentPolynomialBaseConfig attribute), [1] coder_layers (codes.surrogates.LatentNeuralODE.latent_neural_ode_config.LatentNeuralODEBaseConfig attribute), [1] (codes.surrogates.LatentPolynomial.latent_poly_config.LatentPolynomialBaseConfig attribute), [1] coder_width (codes.surrogates.LatentNeuralODE.latent_neural_ode_config.LatentNeuralODEBaseConfig attribute), [1] (codes.surrogates.LatentPolynomial.latent_poly_config.LatentPolynomialBaseConfig attribute), [1] codes module, [1] codes.benchmark module, [1] codes.benchmark.bench_fcts module, [1] codes.benchmark.bench_plots module, [1] codes.benchmark.bench_utils module, [1] codes.surrogates module, [1] codes.surrogates.AbstractSurrogate module, [1] codes.surrogates.AbstractSurrogate.abstract_config module, [1] codes.surrogates.AbstractSurrogate.abstract_surrogate module, [1] codes.surrogates.DeepONet module, [1] codes.surrogates.DeepONet.deeponet module, [1] codes.surrogates.DeepONet.deeponet_config module, [1] codes.surrogates.DeepONet.don_utils module, [1] codes.surrogates.DeepONet.train_utils module, [1] codes.surrogates.FCNN module, [1] codes.surrogates.FCNN.fcnn module, [1] codes.surrogates.FCNN.fcnn_config module, [1] codes.surrogates.LatentNeuralODE module, [1] codes.surrogates.LatentNeuralODE.latent_neural_ode module, [1] codes.surrogates.LatentNeuralODE.latent_neural_ode_config module, [1] codes.surrogates.LatentNeuralODE.utilities module, [1] codes.surrogates.LatentPolynomial module, [1] codes.surrogates.LatentPolynomial.latent_poly module, [1] codes.surrogates.LatentPolynomial.latent_poly_config module, [1] codes.surrogates.surrogate_classes module, [1] codes.train module, [1] codes.train.train_fcts module, [1]

codes.tune module, [1] codes.tune.evaluate_study module, [1] codes.tune.evaluate_tuning module, [1] codes.tune.optuna_fcts module, [1] codes.tune.postgres_fcts module, [1] codes.tune.tune_utils module, [1] codes.utils module, [1] codes.utils.data_utils module, [1] codes.utils.utils module, [1] coeff_layers (codes.surrogates.LatentPolynomial.latent_poly_config.LatentPolynomialBaseConfig attribute), [1] coeff_network (codes.surrogates.LatentPolynomial.latent_poly_config.LatentPolynomialBaseConfig attribute), [1] coeff_width (codes.surrogates.LatentPolynomial.latent_poly_config.LatentPolynomialBaseConfig attribute), [1] compare_batchsize() (in module codes.benchmark), [1] (in module codes.benchmark.bench_fcts), [1] compare_errors() (in module codes.benchmark), [1] (in module codes.benchmark.bench_fcts), [1] compare_extrapolation() (in module codes.benchmark), [1] (in module codes.benchmark.bench_fcts), [1] compare_gradients() (in module codes.benchmark), [1] (in module codes.benchmark.bench_fcts), [1] compare_inference_time() (in module codes.benchmark), [1] (in module codes.benchmark.bench_fcts), [1] compare_interpolation() (in module codes.benchmark), [1] (in module codes.benchmark.bench_fcts), [1] compare_iterative() (in module codes.benchmark.bench_fcts), [1] compare_main_losses() (in module codes.benchmark), [1] (in module codes.benchmark.bench_fcts), [1] compare_models() (in module codes.benchmark), [1] (in module codes.benchmark.bench_fcts), [1] compare_sparse() (in module codes.benchmark), [1] (in module codes.benchmark.bench_fcts), [1] compare_UQ() (in module codes.benchmark), [1] (in module codes.benchmark.bench_fcts), [1] compute_hypervolume_over_time() (in module codes.tune.evaluate_tuning), [1] compute_pareto_plot_data() (in module codes.tune.evaluate_tuning), [1] convert_dict_to_scientific_notation() (in module codes.benchmark), [1] (in module codes.benchmark.bench_utils), [1] convert_to_standard_types() (in module codes.benchmark), [1] (in module codes.benchmark.bench_utils), [1] copy_config() (in module codes.tune), [1] (in module codes.tune.tune_utils), [1] count_trainable_parameters() (in module codes.benchmark), [1] (in module codes.benchmark.bench_utils), [1] create_dataloader() (codes.surrogates.DeepONet.deeponet.MultiONet method), [1] (codes.surrogates.FCNN.fcnn.FullyConnected method), [1] (codes.surrogates.FullyConnected method), [1] (codes.surrogates.LatentNeuralODE method) (codes.surrogates.LatentNeuralODE.latent_neural_ode.LatentNeuralODE method), [1] (codes.surrogates.LatentPoly method), [1] (codes.surrogates.LatentPolynomial.latent_poly.LatentPoly method), [1] (codes.surrogates.MultiONet method), [1] create_dataset() (in module codes.utils), [1] (in module codes.utils.data_utils), [1] create_date_based_directory() (in module codes.surrogates.DeepONet.don_utils), [1] create_hdf5_dataset() (in module codes.utils), [1] (in module codes.utils.data_utils), [1] create_model_dir() (in module codes.utils), [1] (in module codes.utils.utils), [1] create_objective() (in module codes.tune), [1] (in module codes.tune.optuna_fcts), [1] create_task_list_for_surrogate() (in module codes.train), [1] (in module codes.train.train_fcts), [1] custom_collate_fn() (in module codes.surrogates.DeepONet.don_utils), [1]

D

DatasetError, [1] Decoder (class in codes.surrogates), [1] (class in codes.surrogates.LatentNeuralODE.latent_neural_ode), [1] degree (codes.surrogates.LatentPolynomial.latent_poly_config.LatentPolynomialBaseConfig attribute), [1] delete_studies_if_requested() (in module codes.tune), [1] (in module codes.tune.tune_utils), [1] denormalize() (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel method), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel method), [1] (codes.surrogates.AbstractSurrogateModel method), [1] denormalize_old() (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel method), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel method), [1] (codes.surrogates.AbstractSurrogateModel method), [1]

determine_batch_size() (in module codes.utils), [1] (in module codes.utils.utils), [1] discard_numpy_entries() (in module codes.benchmark), [1] (in module codes.benchmark.bench_utils), [1] download_data() (in module codes.utils), [1] (in module codes.utils.data_utils), [1] DownloadProgressBar (class in codes.utils.data_utils), [1] DummyLock (class in codes.train), [1] (class in codes.train.train_fcts), [1]

E

encode_params (codes.surrogates.LatentNeuralODE.latent_neural_ode_config.LatentNeuralODEBaseConfig attribute), [1] Encoder (class in codes.surrogates), [1] (class in codes.surrogates.LatentNeuralODE.latent_neural_ode), [1] epoch() (codes.surrogates.DeepONet.deeponet.MultiONet method), [1] (codes.surrogates.FCNN.fcnn.FullyConnected method), [1] (codes.surrogates.FullyConnected method), [1] (codes.surrogates.MultiONet method), [1] eta_min (codes.surrogates.AbstractSurrogate.abstract_config.AbstractSurrogateBaseConfig attribute), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateBaseConfig attribute), [1] evaluate_accuracy() (in module codes.benchmark), [1] (in module codes.benchmark.bench_fcts), [1] evaluate_batchsize() (in module codes.benchmark), [1] (in module codes.benchmark.bench_fcts), [1]

evaluate_compute() (in module codes.benchmark), [1] (in module codes.benchmark.bench_fcts), [1] evaluate_extrapolation() (in module codes.benchmark), [1] (in module codes.benchmark.bench_fcts), [1] evaluate_gradients() (in module codes.benchmark), [1] (in module codes.benchmark.bench_fcts), [1] evaluate_interpolation() (in module codes.benchmark), [1] (in module codes.benchmark.bench_fcts), [1] evaluate_iterative_predictions() (in module codes.benchmark.bench_fcts), [1] evaluate_sparse() (in module codes.benchmark), [1] (in module codes.benchmark.bench_fcts), [1] evaluate_tuning() (in module codes.tune.evaluate_tuning), [1] evaluate_UQ() (in module codes.benchmark), [1] (in module codes.benchmark.bench_fcts), [1]

F

fc_collate_fn() (in module codes.surrogates.FCNN.fcnn), [1] FCFlatBatchIterable (class in codes.surrogates.FCNN.fcnn), [1] FCNNBaseConfig (class in codes.surrogates.FCNN.fcnn_config), [1] FCPrebatchedDataset (class in codes.surrogates.FCNN.fcnn), [1] first_derivative() (codes.surrogates.LatentNeuralODE.latent_neural_ode.ModelWrapper method), [1] (codes.surrogates.LatentPolynomial.latent_poly.PolynomialModelWrapper method), [1] (codes.surrogates.ModelWrapper method), [1] fit() (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel method), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel method), [1] (codes.surrogates.AbstractSurrogateModel method), [1] (codes.surrogates.DeepONet.deeponet.MultiONet method), [1] (codes.surrogates.FCNN.fcnn.FullyConnected method), [1] (codes.surrogates.FullyConnected method), [1] (codes.surrogates.LatentNeuralODE method) (codes.surrogates.LatentNeuralODE.latent_neural_ode.LatentNeuralODE method), [1] (codes.surrogates.LatentPoly method), [1] (codes.surrogates.LatentPolynomial.latent_poly.LatentPoly method), [1] (codes.surrogates.MultiONet method), [1] fit_profile() (codes.surrogates.LatentNeuralODE method) (codes.surrogates.LatentNeuralODE.latent_neural_ode.LatentNeuralODE method), [1] FlatBatchIterable (class in codes.surrogates.DeepONet.don_utils), [1] FlatSeqBatchIterable (class in codes.surrogates), [1] (class in codes.surrogates.LatentNeuralODE.utilities), [1] flatten_dict() (in module codes.benchmark), [1] (in module codes.benchmark.bench_utils), [1] format_seconds() (in module codes.benchmark), [1] (in module codes.benchmark.bench_utils), [1] format_surrogate_title() (in module codes.tune.evaluate_tuning), [1] format_time() (in module codes.benchmark), [1] (in module codes.benchmark.bench_utils), [1] format_value() (in module codes.benchmark), [1] (in module codes.benchmark.bench_utils), [1] forward() (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel method), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel method), [1] (codes.surrogates.AbstractSurrogateModel method), [1] (codes.surrogates.BranchNet method), [1] (codes.surrogates.Decoder method), [1] (codes.surrogates.DeepONet.deeponet.BranchNet method), [1] (codes.surrogates.DeepONet.deeponet.MultiONet method), [1] (codes.surrogates.DeepONet.deeponet.OperatorNetwork method), [1] (codes.surrogates.DeepONet.deeponet.TrunkNet method), [1] (codes.surrogates.Encoder method), [1] (codes.surrogates.FCNN.fcnn.FullyConnected method), [1] (codes.surrogates.FCNN.fcnn.FullyConnectedNet method), [1] (codes.surrogates.FullyConnected method), [1] (codes.surrogates.FullyConnectedNet method), [1] (codes.surrogates.LatentNeuralODE method) (codes.surrogates.LatentNeuralODE.latent_neural_ode.Decoder method), [1] (codes.surrogates.LatentNeuralODE.latent_neural_ode.Encoder method), [1] (codes.surrogates.LatentNeuralODE.latent_neural_ode.LatentNeuralODE method), [1] (codes.surrogates.LatentNeuralODE.latent_neural_ode.ModelWrapper method), [1] (codes.surrogates.LatentNeuralODE.latent_neural_ode.ODE method), [1] (codes.surrogates.LatentNeuralODE.latent_neural_ode.ODEWithParams method), [1] (codes.surrogates.LatentNeuralODE.latent_neural_ode.OldDecoder method), [1] (codes.surrogates.LatentNeuralODE.latent_neural_ode.OldEncoder method), [1] (codes.surrogates.LatentPoly method), [1] (codes.surrogates.LatentPolynomial.latent_poly.LatentPoly method), [1] (codes.surrogates.LatentPolynomial.latent_poly.OldDecoder method), [1] (codes.surrogates.LatentPolynomial.latent_poly.OldEncoder method), [1] (codes.surrogates.LatentPolynomial.latent_poly.Polynomial method), [1] (codes.surrogates.LatentPolynomial.latent_poly.PolynomialModelWrapper method), [1] (codes.surrogates.ModelWrapper method), [1] (codes.surrogates.MultiONet method), [1] (codes.surrogates.ODE method), [1] (codes.surrogates.Polynomial method), [1] (codes.surrogates.TrunkNet method), [1]

FullyConnected (class in codes.surrogates), [1] (class in codes.surrogates.FCNN.fcnn), [1] FullyConnectedNet (class in codes.surrogates), [1] (class in codes.surrogates.FCNN.fcnn), [1]

G

get_best_trials() (in module codes.tune.evaluate_tuning), [1] get_checkpoint() (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel method), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel method), [1] (codes.surrogates.AbstractSurrogateModel method), [1] get_custom_palette() (in module codes.benchmark), [1] (in module codes.benchmark.bench_plots), [1] get_data_subset() (in module codes.utils), [1] (in module codes.utils.data_utils), [1] get_model_config() (in module codes.benchmark), [1] (in module codes.benchmark.bench_utils), [1]

get_progress_bar() (in module codes.utils), [1] (in module codes.utils.utils), [1] get_project_path() (in module codes.surrogates.DeepONet.don_utils), [1] get_registered_classes() (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel class method), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel class method), [1] (codes.surrogates.AbstractSurrogateModel class method), [1] get_required_models_list() (in module codes.benchmark), [1] (in module codes.benchmark.bench_utils), [1] get_surrogate() (in module codes.benchmark), [1] (in module codes.benchmark.bench_utils), [1]

H

hidden_size (codes.surrogates.DeepONet.deeponet_config.MultiONetBaseConfig attribute), [1] (codes.surrogates.FCNN.fcnn_config.FCNNBaseConfig attribute), [1]

hypervolume_2d() (in module codes.tune.evaluate_tuning), [1]

I

identity_loss() (codes.surrogates.LatentNeuralODE.latent_neural_ode.ModelWrapper method), [1] (codes.surrogates.LatentPolynomial.latent_poly.PolynomialModelWrapper method), [1] (codes.surrogates.ModelWrapper method), [1]

inference_time_bar_plot() (in module codes.benchmark), [1] (in module codes.benchmark.bench_plots), [1] initialize_optuna_database() (in module codes.tune), [1] (in module codes.tune.postgres_fcts), [1]

L

latent_features (codes.surrogates.LatentNeuralODE.latent_neural_ode_config.LatentNeuralODEBaseConfig attribute), [1] (codes.surrogates.LatentPolynomial.latent_poly_config.LatentPolynomialBaseConfig attribute), [1] LatentNeuralODE (class in codes.surrogates) (class in codes.surrogates.LatentNeuralODE.latent_neural_ode), [1] LatentNeuralODEBaseConfig (class in codes.surrogates.LatentNeuralODE.latent_neural_ode_config), [1] LatentPoly (class in codes.surrogates), [1] (class in codes.surrogates.LatentPolynomial.latent_poly), [1] LatentPolynomialBaseConfig (class in codes.surrogates.LatentPolynomial.latent_poly_config), [1] layers_factor (codes.surrogates.LatentNeuralODE.latent_neural_ode_config.LatentNeuralODEBaseConfig attribute), [1] (codes.surrogates.LatentPolynomial.latent_poly_config.LatentPolynomialBaseConfig attribute), [1] learning_rate (codes.surrogates.AbstractSurrogate.abstract_config.AbstractSurrogateBaseConfig attribute), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateBaseConfig attribute), [1] list_pth_files() (in module codes.surrogates.DeepONet.don_utils), [1] load() (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel method), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel method), [1] (codes.surrogates.AbstractSurrogateModel method), [1]

load_and_save_config() (in module codes.utils), [1] (in module codes.utils.utils), [1] load_loss_history() (in module codes.tune.evaluate_tuning), [1] load_model() (in module codes.benchmark), [1] (in module codes.benchmark.bench_utils), [1] load_model_test_losses() (in module codes.tune), [1] (in module codes.tune.evaluate_study), [1] load_study_config() (in module codes.tune), [1] (in module codes.tune.evaluate_study), [1] load_task_list() (in module codes.utils), [1] (in module codes.utils.utils), [1] load_yaml_config() (in module codes.tune), [1] (in module codes.tune.optuna_fcts), [1] loss (codes.surrogates.AbstractSurrogate.abstract_config.AbstractSurrogateBaseConfig property), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateBaseConfig property), [1] loss_function (codes.surrogates.AbstractSurrogate.abstract_config.AbstractSurrogateBaseConfig attribute), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateBaseConfig attribute), [1]

M

main() (in module codes.tune.evaluate_study), [1] (in module codes.tune.evaluate_tuning), [1] make_comparison_csv() (in module codes.benchmark), [1] (in module codes.benchmark.bench_utils), [1] make_description() (in module codes.utils), [1] (in module codes.utils.utils), [1] make_optuna_params() (in module codes.tune), [1] (in module codes.tune.optuna_fcts), [1] mass_conservation_loss() (in module codes.surrogates.DeepONet.don_utils), [1] (in module codes.surrogates.DeepONet.train_utils), [1] masses (codes.surrogates.DeepONet.deeponet_config.MultiONetBaseConfig attribute), [1] massloss_factor (codes.surrogates.DeepONet.deeponet_config.MultiONetBaseConfig attribute), [1] MaxValidTrialsCallback (class in codes.tune), [1] (class in codes.tune.optuna_fcts), [1] maybe_set_runtime_threshold() (in module codes.tune), [1] (in module codes.tune.optuna_fcts), [1] measure_inference_time() (in module codes.benchmark), [1] (in module codes.benchmark.bench_utils), [1] measure_memory_footprint() (in module codes.benchmark), [1] (in module codes.benchmark.bench_utils), [1] model_version (codes.surrogates.LatentNeuralODE.latent_neural_ode_config.LatentNeuralODEBaseConfig attribute), [1] (codes.surrogates.LatentPolynomial.latent_poly_config.LatentPolynomialBaseConfig attribute), [1] ModelWrapper (class in codes.surrogates), [1] (class in codes.surrogates.LatentNeuralODE.latent_neural_ode), [1] module codes, [1] codes.benchmark, [1] codes.benchmark.bench_fcts, [1] codes.benchmark.bench_plots, [1] codes.benchmark.bench_utils, [1] codes.surrogates, [1] codes.surrogates.AbstractSurrogate, [1] codes.surrogates.AbstractSurrogate.abstract_config, [1] codes.surrogates.AbstractSurrogate.abstract_surrogate, [1] codes.surrogates.DeepONet, [1] codes.surrogates.DeepONet.deeponet, [1] codes.surrogates.DeepONet.deeponet_config, [1] codes.surrogates.DeepONet.don_utils, [1] codes.surrogates.DeepONet.train_utils, [1] codes.surrogates.FCNN, [1] codes.surrogates.FCNN.fcnn, [1] codes.surrogates.FCNN.fcnn_config, [1] codes.surrogates.LatentNeuralODE, [1] codes.surrogates.LatentNeuralODE.latent_neural_ode, [1] codes.surrogates.LatentNeuralODE.latent_neural_ode_config, [1] codes.surrogates.LatentNeuralODE.utilities, [1] codes.surrogates.LatentPolynomial, [1] codes.surrogates.LatentPolynomial.latent_poly, [1] codes.surrogates.LatentPolynomial.latent_poly_config, [1] codes.surrogates.surrogate_classes, [1] codes.train, [1] codes.train.train_fcts, [1] codes.tune, [1] codes.tune.evaluate_study, [1] codes.tune.evaluate_tuning, [1] codes.tune.optuna_fcts, [1] codes.tune.postgres_fcts, [1] codes.tune.tune_utils, [1] codes.utils, [1] codes.utils.data_utils, [1] codes.utils.utils, [1]

momentum (codes.surrogates.AbstractSurrogate.abstract_config.AbstractSurrogateBaseConfig attribute), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateBaseConfig attribute), [1] moving_average() (in module codes.tune), [1] (in module codes.tune.evaluate_study), [1] MultiONet (class in codes.surrogates), [1] (class in codes.surrogates.DeepONet.deeponet), [1] MultiONetBaseConfig (class in codes.surrogates.DeepONet.deeponet_config), [1]

N

nice_print() (in module codes.utils), [1] (in module codes.utils.utils), [1]

normalize_data() (in module codes.utils), [1] (in module codes.utils.data_utils), [1] num_hidden_layers (codes.surrogates.FCNN.fcnn_config.FCNNBaseConfig attribute), [1]

O

ODE (class in codes.surrogates), [1] (class in codes.surrogates.LatentNeuralODE.latent_neural_ode), [1] ode_layers (codes.surrogates.LatentNeuralODE.latent_neural_ode_config.LatentNeuralODEBaseConfig attribute), [1] ode_tanh_reg (codes.surrogates.LatentNeuralODE.latent_neural_ode_config.LatentNeuralODEBaseConfig attribute), [1] ode_width (codes.surrogates.LatentNeuralODE.latent_neural_ode_config.LatentNeuralODEBaseConfig attribute), [1] ODEWithParams (class in codes.surrogates.LatentNeuralODE.latent_neural_ode), [1] OldDecoder (class in codes.surrogates.LatentNeuralODE.latent_neural_ode), [1] (class in codes.surrogates.LatentPolynomial.latent_poly), [1]

OldEncoder (class in codes.surrogates.LatentNeuralODE.latent_neural_ode), [1] (class in codes.surrogates.LatentPolynomial.latent_poly), [1] OperatorNetwork (class in codes.surrogates.DeepONet.deeponet), [1] optimizer (codes.surrogates.AbstractSurrogate.abstract_config.AbstractSurrogateBaseConfig attribute), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateBaseConfig attribute), [1] output_factor (codes.surrogates.DeepONet.deeponet_config.MultiONetBaseConfig attribute), [1]

P

parallel_training() (in module codes.train), [1] (in module codes.train.train_fcts), [1] params_branch (codes.surrogates.DeepONet.deeponet_config.MultiONetBaseConfig attribute), [1] pareto_front() (in module codes.tune.evaluate_tuning), [1] pareto_front_mask() (in module codes.tune.evaluate_tuning), [1] parse_args() (in module codes.tune.evaluate_tuning), [1] parse_arguments() (in module codes.tune.evaluate_study), [1] parse_for_none() (in module codes.utils.utils), [1] parse_hyperparameters() (in module codes.utils), [1] (in module codes.utils.utils), [1] plot_all_generalization_errors() (in module codes.benchmark), [1] (in module codes.benchmark.bench_plots), [1] plot_average_errors_over_time() (in module codes.benchmark), [1] (in module codes.benchmark.bench_plots), [1] plot_average_uncertainty_over_time() (in module codes.benchmark), [1] (in module codes.benchmark.bench_plots), [1] plot_catastrophic_detection_curves() (in module codes.benchmark.bench_plots), [1] plot_comparative_error_correlation_heatmaps() (in module codes.benchmark), [1] (in module codes.benchmark.bench_plots), [1] plot_comparative_gradient_heatmaps() (in module codes.benchmark), [1] (in module codes.benchmark.bench_plots), [1] plot_dynamic_correlation() (in module codes.benchmark), [1] (in module codes.benchmark.bench_plots), [1] plot_error_distribution_comparative() (in module codes.benchmark), [1] (in module codes.benchmark.bench_plots), [1] plot_error_distribution_per_quantity() (in module codes.benchmark), [1] (in module codes.benchmark.bench_plots), [1] plot_error_percentiles_over_time() (in module codes.benchmark), [1] (in module codes.benchmark.bench_plots), [1] plot_errors_over_time() (in module codes.benchmark.bench_plots), [1] plot_example_iterative_predictions() (in module codes.benchmark), [1] (in module codes.benchmark.bench_plots), [1] plot_example_mode_predictions() (in module codes.benchmark), [1] (in module codes.benchmark.bench_plots), [1] plot_example_predictions_with_uncertainty() (in module codes.benchmark), [1] (in module codes.benchmark.bench_plots), [1] plot_generalization_error_comparison() (in module codes.benchmark), [1] (in module codes.benchmark.bench_plots), [1] plot_generalization_errors() (in module codes.benchmark), [1] (in module codes.benchmark.bench_plots), [1] plot_gradients_heatmap() (in module codes.benchmark), [1] (in module codes.benchmark.bench_plots), [1] plot_loss_comparison() (in module codes.benchmark), [1] (in module codes.benchmark.bench_plots), [1] plot_loss_comparison_equal() (in module codes.benchmark.bench_plots), [1] plot_loss_comparison_train_duration() (in module codes.benchmark), [1] (in module codes.benchmark.bench_plots), [1]

plot_losses() (in module codes.benchmark), [1] (in module codes.benchmark.bench_plots), [1] (in module codes.tune.evaluate_tuning), [1] plot_losses_dual_axis() (in module codes.benchmark.bench_plots), [1] plot_MAE_comparison() (in module codes.benchmark), [1] (in module codes.benchmark.bench_plots), [1] plot_mean_deltadex_over_time_main_vs_ensemble() (in module codes.benchmark.bench_plots), [1] plot_relative_errors() (in module codes.benchmark), [1] (in module codes.benchmark.bench_plots), [1] plot_surr_losses() (in module codes.benchmark), [1] (in module codes.benchmark.bench_plots), [1] plot_test_losses() (in module codes.tune), [1] (in module codes.tune.evaluate_study), [1] plot_uncertainty_confidence() (in module codes.benchmark), [1] (in module codes.benchmark.bench_plots), [1] plot_uncertainty_heatmap() (in module codes.benchmark), [1] (in module codes.benchmark.bench_plots), [1] plot_uncertainty_over_time_comparison() (in module codes.benchmark), [1] (in module codes.benchmark.bench_plots), [1] plot_uncertainty_vs_errors() (in module codes.benchmark), [1] (in module codes.benchmark.bench_plots), [1] poly_power (codes.surrogates.AbstractSurrogate.abstract_config.AbstractSurrogateBaseConfig attribute), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateBaseConfig attribute), [1] Polynomial (class in codes.surrogates), [1] (class in codes.surrogates.LatentPolynomial.latent_poly), [1] PolynomialModelWrapper (class in codes.surrogates.LatentPolynomial.latent_poly), [1] post_init_check() (codes.surrogates.DeepONet.deeponet.OperatorNetwork method), [1] PreBatchedDataset (class in codes.surrogates.DeepONet.don_utils), [1] predict() (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel method), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel method), [1] (codes.surrogates.AbstractSurrogateModel method), [1] prepare_data() (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel method), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel method), [1] (codes.surrogates.AbstractSurrogateModel method), [1] (codes.surrogates.DeepONet.deeponet.MultiONet method), [1] (codes.surrogates.FCNN.fcnn.FullyConnected method), [1] (codes.surrogates.FullyConnected method), [1] (codes.surrogates.LatentNeuralODE method) (codes.surrogates.LatentNeuralODE.latent_neural_ode.LatentNeuralODE method), [1] (codes.surrogates.LatentPoly method), [1] (codes.surrogates.LatentPolynomial.latent_poly.LatentPoly method), [1] (codes.surrogates.MultiONet method), [1] prepare_workspace() (in module codes.tune), [1] (in module codes.tune.tune_utils), [1] print_data_info() (in module codes.utils.data_utils), [1]

R

read_yaml_config() (in module codes.benchmark), [1] (in module codes.surrogates.DeepONet.don_utils), [1] (in module codes.utils), [1] (in module codes.utils.utils), [1] register() (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel class method), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel class method), [1] (codes.surrogates.AbstractSurrogateModel class method), [1] regularization_factor (codes.surrogates.AbstractSurrogate.abstract_config.AbstractSurrogateBaseConfig attribute), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateBaseConfig attribute), [1]

rel_errors_and_uq() (in module codes.benchmark), [1] (in module codes.benchmark.bench_plots), [1] release() (codes.train.DummyLock method), [1] (codes.train.train_fcts.DummyLock method), [1] rtol (codes.surrogates.LatentNeuralODE.latent_neural_ode_config.LatentNeuralODEBaseConfig attribute), [1] run_benchmark() (in module codes.benchmark), [1] (in module codes.benchmark.bench_fcts), [1]

S

save() (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel method), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel method), [1] (codes.surrogates.AbstractSurrogateModel method), [1] save_individual_pareto_plot() (in module codes.tune.evaluate_tuning), [1] save_pareto_front_grid() (in module codes.tune.evaluate_tuning), [1] save_plot() (in module codes.benchmark), [1] (in module codes.benchmark.bench_plots), [1] save_plot_counter() (in module codes.benchmark), [1] (in module codes.benchmark.bench_plots), [1] (in module codes.surrogates.DeepONet.don_utils), [1] save_relative_hv_combined() (in module codes.tune.evaluate_tuning), [1] save_relative_hv_grid() (in module codes.tune.evaluate_tuning), [1] save_table_csv() (in module codes.benchmark), [1] (in module codes.benchmark.bench_utils), [1] save_task_list() (in module codes.utils), [1] (in module codes.utils.utils), [1] scheduler (codes.surrogates.AbstractSurrogate.abstract_config.AbstractSurrogateBaseConfig attribute), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateBaseConfig attribute), [1] second_derivative() (codes.surrogates.LatentNeuralODE.latent_neural_ode.ModelWrapper method), [1] (codes.surrogates.LatentPolynomial.latent_poly.PolynomialModelWrapper method), [1] (codes.surrogates.ModelWrapper method), [1]

sequential_training() (in module codes.train), [1] (in module codes.train.train_fcts), [1] set_params() (codes.surrogates.LatentNeuralODE.latent_neural_ode.ODEWithParams method), [1] set_random_seed() (in module codes.surrogates.DeepONet.don_utils), [1] set_random_seeds() (in module codes.utils), [1] (in module codes.utils.utils), [1] setup_checkpoint() (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel method), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel method), [1] (codes.surrogates.AbstractSurrogateModel method), [1] setup_criterion() (codes.surrogates.DeepONet.deeponet.MultiONet method), [1] (codes.surrogates.MultiONet method), [1] setup_optimizer_and_scheduler() (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel method), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel method), [1] (codes.surrogates.AbstractSurrogateModel method), [1] setup_progress_bar() (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel method), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel method), [1] (codes.surrogates.AbstractSurrogateModel method), [1] summarize_pareto_tradeoff() (in module codes.tune.evaluate_tuning), [1]

T

tabular_comparison() (in module codes.benchmark), [1] (in module codes.benchmark.bench_fcts), [1] time_execution() (in module codes.utils), [1] (in module codes.utils.utils), [1] time_inference() (in module codes.benchmark), [1] (in module codes.benchmark.bench_fcts), [1] time_pruning() (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel method), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel method), [1] (codes.surrogates.AbstractSurrogateModel method), [1] total_loss() (codes.surrogates.LatentNeuralODE.latent_neural_ode.ModelWrapper method), [1] (codes.surrogates.LatentPolynomial.latent_poly.PolynomialModelWrapper method), [1] (codes.surrogates.ModelWrapper method), [1]

train_and_save_model() (in module codes.train), [1] (in module codes.train.train_fcts), [1] training_run() (in module codes.tune), [1] (in module codes.tune.optuna_fcts), [1] trunk_hidden_layers (codes.surrogates.DeepONet.deeponet_config.MultiONetBaseConfig attribute), [1] trunk_input_size (codes.surrogates.DeepONet.deeponet_config.MultiONetBaseConfig attribute), [1] TrunkNet (class in codes.surrogates), [1] (class in codes.surrogates.DeepONet.deeponet), [1]

U

update_to() (codes.utils.data_utils.DownloadProgressBar method), [1]

V

validate() (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel method), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel method), [1] (codes.surrogates.AbstractSurrogateModel method), [1]

W

worker() (in module codes.train), [1] (in module codes.train.train_fcts), [1] worker_init_fn() (in module codes.utils), [1] (in module codes.utils.utils), [1]

write_metrics_to_yaml() (in module codes.benchmark), [1] (in module codes.benchmark.bench_utils), [1]

Y

yes_no() (in module codes.tune), [1] (in module codes.tune.tune_utils), [1]