Index A | B | C | D | E | F | G | I | L | M | N | O | P | R | S | T | U | V | W | Y A AbstractSurrogateBaseConfig (class in codes.surrogates.AbstractSurrogate) (class in codes.surrogates.AbstractSurrogate.abstract_config) AbstractSurrogateModel (class in codes.surrogates) (class in codes.surrogates.AbstractSurrogate) (class in codes.surrogates.AbstractSurrogate.abstract_surrogate) acquire() (codes.train.DummyLock method) (codes.train.train_fcts.DummyLock method) activation (codes.surrogates.AbstractSurrogate.abstract_config.AbstractSurrogateBaseConfig attribute), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateBaseConfig attribute), [1] B batch_factor_to_float() (in module codes.utils) (in module codes.utils.utils) beta (codes.surrogates.AbstractSurrogate.abstract_config.AbstractSurrogateBaseConfig attribute) (codes.surrogates.AbstractSurrogate.AbstractSurrogateBaseConfig attribute) BranchNet (class in codes.surrogates) build_study_names() (in module codes.tune) (in module codes.tune.tune_utils) C check_and_load_data() (in module codes.utils) (in module codes.utils.data_utils) check_benchmark() (in module codes.benchmark) (in module codes.benchmark.bench_utils) check_surrogate() (in module codes.benchmark) (in module codes.benchmark.bench_utils) check_training_status() (in module codes.utils) (in module codes.utils.utils) checkpoint() (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel method) (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel method) (codes.surrogates.AbstractSurrogateModel method) ChemDataset (class in codes.surrogates) clean_metrics() (in module codes.benchmark) (in module codes.benchmark.bench_utils) codes module 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.surrogate_classes module 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] coef (codes.surrogates.Polynomial attribute) compare_batchsize() (in module codes.benchmark) (in module codes.benchmark.bench_fcts) compare_dynamic_accuracy() (in module codes.benchmark) (in module codes.benchmark.bench_fcts) compare_extrapolation() (in module codes.benchmark) (in module codes.benchmark.bench_fcts) compare_inference_time() (in module codes.benchmark) (in module codes.benchmark.bench_fcts) compare_interpolation() (in module codes.benchmark) (in module codes.benchmark.bench_fcts) compare_main_losses() (in module codes.benchmark) (in module codes.benchmark.bench_fcts) compare_models() (in module codes.benchmark) (in module codes.benchmark.bench_fcts) compare_relative_errors() (in module codes.benchmark) (in module codes.benchmark.bench_fcts) compare_sparse() (in module codes.benchmark) (in module codes.benchmark.bench_fcts) compare_UQ() (in module codes.benchmark) (in module codes.benchmark.bench_fcts) config (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel attribute) (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel attribute) (codes.surrogates.AbstractSurrogateModel attribute) (codes.surrogates.LatentPoly attribute) convert_dict_to_scientific_notation() (in module codes.benchmark) (in module codes.benchmark.bench_utils) convert_to_standard_types() (in module codes.benchmark) (in module codes.benchmark.bench_utils) copy_config() (in module codes.tune) (in module codes.tune.tune_utils) count_trainable_parameters() (in module codes.benchmark) (in module codes.benchmark.bench_utils) create_dataloader() (codes.surrogates.FullyConnected method) (codes.surrogates.LatentNeuralODE method) (codes.surrogates.LatentPoly method) (codes.surrogates.MultiONet method) create_dataset() (in module codes.utils) (in module codes.utils.data_utils) create_hdf5_dataset() (in module codes.utils) (in module codes.utils.data_utils) create_model_dir() (in module codes.utils) (in module codes.utils.utils) create_objective() (in module codes.tune) (in module codes.tune.optuna_fcts) create_task_list_for_surrogate() (in module codes.train) (in module codes.train.train_fcts) D DatasetError Decoder (class in codes.surrogates) degree (codes.surrogates.Polynomial attribute) delete_studies_if_requested() (in module codes.tune) (in module codes.tune.tune_utils) denormalize() (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel method) (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel method) (codes.surrogates.AbstractSurrogateModel method) denormalize_old() (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel method) (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel method) (codes.surrogates.AbstractSurrogateModel method) determine_batch_size() (in module codes.utils) (in module codes.utils.utils) device (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel attribute) (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel attribute) (codes.surrogates.AbstractSurrogateModel attribute) (codes.surrogates.LatentPoly attribute) dimension (codes.surrogates.Polynomial attribute) discard_numpy_entries() (in module codes.benchmark) (in module codes.benchmark.bench_utils) download_data() (in module codes.utils) (in module codes.utils.data_utils) DownloadProgressBar (class in codes.utils.data_utils) DummyLock (class in codes.train) (class in codes.train.train_fcts) E Encoder (class in codes.surrogates) epoch() (codes.surrogates.FullyConnected method) (codes.surrogates.MultiONet method) eta_min (codes.surrogates.AbstractSurrogate.abstract_config.AbstractSurrogateBaseConfig attribute), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateBaseConfig attribute), [1] evaluate_accuracy() (in module codes.benchmark) (in module codes.benchmark.bench_fcts) evaluate_batchsize() (in module codes.benchmark) (in module codes.benchmark.bench_fcts) evaluate_compute() (in module codes.benchmark) (in module codes.benchmark.bench_fcts) evaluate_dynamic_accuracy() (in module codes.benchmark) (in module codes.benchmark.bench_fcts) evaluate_extrapolation() (in module codes.benchmark) (in module codes.benchmark.bench_fcts) evaluate_interpolation() (in module codes.benchmark) (in module codes.benchmark.bench_fcts) evaluate_iterative_predictions() (in module codes.benchmark.bench_fcts) evaluate_sparse() (in module codes.benchmark) (in module codes.benchmark.bench_fcts) evaluate_tuning() (in module codes.tune.evaluate_tuning) evaluate_UQ() (in module codes.benchmark) (in module codes.benchmark.bench_fcts) F first_derivative() (codes.surrogates.ModelWrapper method) fit() (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel method) (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel method) (codes.surrogates.AbstractSurrogateModel method) (codes.surrogates.FullyConnected method) (codes.surrogates.LatentNeuralODE method) (codes.surrogates.LatentPoly method) (codes.surrogates.MultiONet method) fit_profile() (codes.surrogates.LatentNeuralODE method) FlatSeqBatchIterable (class in codes.surrogates) flatten_dict() (in module codes.benchmark) (in module codes.benchmark.bench_utils) format_seconds() (in module codes.benchmark) (in module codes.benchmark.bench_utils) format_time() (in module codes.benchmark) (in module codes.benchmark.bench_utils) forward() (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel method) (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel method) (codes.surrogates.AbstractSurrogateModel method) (codes.surrogates.BranchNet method) (codes.surrogates.Decoder method) (codes.surrogates.Encoder method) (codes.surrogates.FullyConnected method) (codes.surrogates.FullyConnectedNet method) (codes.surrogates.LatentNeuralODE method) (codes.surrogates.LatentPoly method) (codes.surrogates.ModelWrapper method) (codes.surrogates.MultiONet method) (codes.surrogates.ODE method) (codes.surrogates.Polynomial method) (codes.surrogates.TrunkNet method) FullyConnected (class in codes.surrogates) FullyConnectedNet (class in codes.surrogates) G get_best_trials() (in module codes.tune.evaluate_tuning) get_checkpoint() (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel method) (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel method) (codes.surrogates.AbstractSurrogateModel method) get_custom_palette() (in module codes.benchmark) (in module codes.benchmark.bench_plots) get_data_subset() (in module codes.utils) (in module codes.utils.data_utils) get_model_config() (in module codes.benchmark) (in module codes.benchmark.bench_utils) get_progress_bar() (in module codes.utils) (in module codes.utils.utils) get_registered_classes() (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel class method) (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel class method) (codes.surrogates.AbstractSurrogateModel class method) get_required_models_list() (in module codes.benchmark) (in module codes.benchmark.bench_utils) get_surrogate() (in module codes.benchmark) (in module codes.benchmark.bench_utils) I identity_loss() (codes.surrogates.ModelWrapper method) inference_time_bar_plot() (in module codes.benchmark) (in module codes.benchmark.bench_plots) initialize_optuna_database() (in module codes.tune) (in module codes.tune.postgres_fcts) int_ext_sparse() (in module codes.benchmark) (in module codes.benchmark.bench_plots) L L1 (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel attribute) (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel attribute) (codes.surrogates.AbstractSurrogateModel attribute) LatentNeuralODE (class in codes.surrogates) LatentPoly (class in codes.surrogates) learning_rate (codes.surrogates.AbstractSurrogate.abstract_config.AbstractSurrogateBaseConfig attribute), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateBaseConfig attribute), [1] load() (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel method) (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel method) (codes.surrogates.AbstractSurrogateModel method) load_and_save_config() (in module codes.utils) (in module codes.utils.utils) load_loss_history() (in module codes.tune.evaluate_tuning) load_model() (in module codes.benchmark) (in module codes.benchmark.bench_utils) load_model_test_losses() (in module codes.tune) (in module codes.tune.evaluate_study) load_study_config() (in module codes.tune) (in module codes.tune.evaluate_study) load_task_list() (in module codes.utils) (in module codes.utils.utils) load_yaml_config() (in module codes.tune) (in module codes.tune.optuna_fcts) loss (codes.surrogates.AbstractSurrogate.abstract_config.AbstractSurrogateBaseConfig property) (codes.surrogates.AbstractSurrogate.AbstractSurrogateBaseConfig property) loss_function (codes.surrogates.AbstractSurrogate.abstract_config.AbstractSurrogateBaseConfig attribute), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateBaseConfig attribute), [1] loss_kwargs (codes.surrogates.AbstractSurrogate.abstract_config.AbstractSurrogateBaseConfig attribute) (codes.surrogates.AbstractSurrogate.AbstractSurrogateBaseConfig attribute) M MAE (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel attribute) (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel attribute) (codes.surrogates.AbstractSurrogateModel attribute) main() (in module codes.tune.evaluate_study) (in module codes.tune.evaluate_tuning) make_comparison_csv() (in module codes.benchmark) (in module codes.benchmark.bench_utils) make_description() (in module codes.utils) (in module codes.utils.utils) make_optuna_params() (in module codes.tune) (in module codes.tune.optuna_fcts) maybe_set_runtime_threshold() (in module codes.tune) (in module codes.tune.optuna_fcts) measure_inference_time() (in module codes.benchmark) (in module codes.benchmark.bench_utils) measure_memory_footprint() (in module codes.benchmark) (in module codes.benchmark.bench_utils) model (codes.surrogates.LatentPoly attribute) ModelWrapper (class in codes.surrogates) module codes 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.surrogate_classes 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) (in module codes.tune.evaluate_study) MultiONet (class in codes.surrogates) N n_quantities (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel attribute) (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel attribute) (codes.surrogates.AbstractSurrogateModel attribute) n_timesteps (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel attribute) (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel attribute) (codes.surrogates.AbstractSurrogateModel attribute) nice_print() (in module codes.utils) (in module codes.utils.utils) normalisation (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel attribute) (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel attribute) (codes.surrogates.AbstractSurrogateModel attribute) normalize_data() (in module codes.utils) (in module codes.utils.data_utils) O ODE (class in codes.surrogates) optimizer (codes.surrogates.AbstractSurrogate.abstract_config.AbstractSurrogateBaseConfig attribute), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateBaseConfig attribute), [1] P parallel_training() (in module codes.train) (in module codes.train.train_fcts) parse_args() (in module codes.tune.evaluate_tuning) parse_arguments() (in module codes.tune.evaluate_study) parse_for_none() (in module codes.utils.utils) plot_all_generalization_errors() (in module codes.benchmark) (in module codes.benchmark.bench_plots) plot_average_errors_over_time() (in module codes.benchmark) (in module codes.benchmark.bench_plots) plot_average_uncertainty_over_time() (in module codes.benchmark) (in module codes.benchmark.bench_plots) plot_comparative_dynamic_correlation_heatmaps() (in module codes.benchmark) (in module codes.benchmark.bench_plots) plot_comparative_error_correlation_heatmaps() (in module codes.benchmark) (in module codes.benchmark.bench_plots) plot_dynamic_correlation() (in module codes.benchmark) (in module codes.benchmark.bench_plots) plot_dynamic_correlation_heatmap() (in module codes.benchmark) (in module codes.benchmark.bench_plots) plot_dynamic_correlation_heatmap_old() (in module codes.benchmark.bench_plots) plot_error_correlation_heatmap() (in module codes.benchmark) (in module codes.benchmark.bench_plots) plot_error_correlation_heatmap_old() (in module codes.benchmark.bench_plots) plot_error_distribution_comparative() (in module codes.benchmark) (in module codes.benchmark.bench_plots) plot_error_distribution_per_quantity() (in module codes.benchmark) (in module codes.benchmark.bench_plots) plot_error_percentiles_over_time() (in module codes.benchmark) (in module codes.benchmark.bench_plots) plot_example_iterative_predictions() (in module codes.benchmark) (in module codes.benchmark.bench_plots) plot_example_mode_predictions() (in module codes.benchmark) (in module codes.benchmark.bench_plots) plot_example_predictions_with_uncertainty() (in module codes.benchmark) (in module codes.benchmark.bench_plots) plot_generalization_error_comparison() (in module codes.benchmark) (in module codes.benchmark.bench_plots) plot_generalization_errors() (in module codes.benchmark) (in module codes.benchmark.bench_plots) plot_loss_comparison() (in module codes.benchmark) (in module codes.benchmark.bench_plots) plot_loss_comparison_equal() (in module codes.benchmark.bench_plots) plot_loss_comparison_train_duration() (in module codes.benchmark) (in module codes.benchmark.bench_plots) plot_losses() (in module codes.benchmark) (in module codes.benchmark.bench_plots) (in module codes.tune.evaluate_tuning) plot_losses_dual_axis() (in module codes.benchmark.bench_plots) plot_MAE_comparison() (in module codes.benchmark) (in module codes.benchmark.bench_plots) plot_relative_errors() (in module codes.benchmark) (in module codes.benchmark.bench_plots) plot_surr_losses() (in module codes.benchmark) (in module codes.benchmark.bench_plots) plot_test_losses() (in module codes.tune) (in module codes.tune.evaluate_study) plot_uncertainty_confidence() (in module codes.benchmark) (in module codes.benchmark.bench_plots) plot_uncertainty_over_time_comparison() (in module codes.benchmark) (in module codes.benchmark.bench_plots) plot_uncertainty_vs_errors() (in module codes.benchmark) (in module codes.benchmark.bench_plots) poly_power (codes.surrogates.AbstractSurrogate.abstract_config.AbstractSurrogateBaseConfig attribute), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateBaseConfig attribute), [1] Polynomial (class in codes.surrogates) predict() (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel method) (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel method) (codes.surrogates.AbstractSurrogateModel method) prepare_data() (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel method) (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel method) (codes.surrogates.AbstractSurrogateModel method) (codes.surrogates.FullyConnected method) (codes.surrogates.LatentNeuralODE method) (codes.surrogates.LatentPoly method) (codes.surrogates.MultiONet method) prepare_workspace() (in module codes.tune) (in module codes.tune.tune_utils) print_data_info() (in module codes.utils.data_utils) R read_yaml_config() (in module codes.benchmark) (in module codes.utils) (in module codes.utils.utils) register() (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel class method) (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel class method) (codes.surrogates.AbstractSurrogateModel class method) regularization_factor (codes.surrogates.AbstractSurrogate.abstract_config.AbstractSurrogateBaseConfig attribute), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateBaseConfig attribute), [1] rel_errors_and_uq() (in module codes.benchmark) (in module codes.benchmark.bench_plots) release() (codes.train.DummyLock method) (codes.train.train_fcts.DummyLock method) run_benchmark() (in module codes.benchmark) (in module codes.benchmark.bench_fcts) S save() (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel method) (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel method) (codes.surrogates.AbstractSurrogateModel method) save_plot() (in module codes.benchmark) (in module codes.benchmark.bench_plots) save_plot_counter() (in module codes.benchmark) (in module codes.benchmark.bench_plots) save_table_csv() (in module codes.benchmark) (in module codes.benchmark.bench_utils) save_task_list() (in module codes.utils) (in module codes.utils.utils) scheduler (codes.surrogates.AbstractSurrogate.abstract_config.AbstractSurrogateBaseConfig attribute), [1] (codes.surrogates.AbstractSurrogate.AbstractSurrogateBaseConfig attribute), [1] second_derivative() (codes.surrogates.ModelWrapper method) sequential_training() (in module codes.train) (in module codes.train.train_fcts) set_random_seeds() (in module codes.utils) (in module codes.utils.utils) setup_checkpoint() (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel method) (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel method) (codes.surrogates.AbstractSurrogateModel method) setup_criterion() (codes.surrogates.MultiONet method) setup_optimizer_and_scheduler() (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel method) (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel method) (codes.surrogates.AbstractSurrogateModel method) setup_progress_bar() (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel method) (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel method) (codes.surrogates.AbstractSurrogateModel method) surrogate_classes (in module codes.surrogates) T t_matrix (codes.surrogates.Polynomial attribute) tabular_comparison() (in module codes.benchmark) (in module codes.benchmark.bench_fcts) test_loss (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel attribute) (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel attribute) (codes.surrogates.AbstractSurrogateModel attribute) time_execution() (in module codes.utils) (in module codes.utils.utils) time_inference() (in module codes.benchmark) (in module codes.benchmark.bench_fcts) time_pruning() (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel method) (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel method) (codes.surrogates.AbstractSurrogateModel method) total_loss() (codes.surrogates.ModelWrapper method) train_and_save_model() (in module codes.train) (in module codes.train.train_fcts) train_duration (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel attribute) (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel attribute) (codes.surrogates.AbstractSurrogateModel attribute) train_loss (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel attribute) (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel attribute) (codes.surrogates.AbstractSurrogateModel attribute) training_run() (in module codes.tune) (in module codes.tune.optuna_fcts) TrunkNet (class in codes.surrogates) U update_to() (codes.utils.data_utils.DownloadProgressBar method) V validate() (codes.surrogates.AbstractSurrogate.abstract_surrogate.AbstractSurrogateModel method) (codes.surrogates.AbstractSurrogate.AbstractSurrogateModel method) (codes.surrogates.AbstractSurrogateModel method) W worker() (in module codes.train) (in module codes.train.train_fcts) worker_init_fn() (in module codes.utils) (in module codes.utils.utils) write_metrics_to_yaml() (in module codes.benchmark) (in module codes.benchmark.bench_utils) Y yes_no() (in module codes.tune) (in module codes.tune.tune_utils)
