Index

A

ACW (class in Models.neural_models)

AFNONet (class in Models.AFNO)

B

BNN (class in Models.neural_models)

C

CGST (class in Models.GST) compute_and_sample_values() (in module Data.Synthetic.generate_synthetic_data) compute_covariance() (in module Data.Synthetic.synthetic_data_util) compute_mean() (in module Data.Synthetic.synthetic_data_util)

compute_output() (in module Data.Synthetic.synthetic_data_util) ConfigManager (class in Results.parser) create_GridStationDataset() (in module Data.forecast_dataset) create_model() (in module Results.extract_from_configuration)

D

Data.data_losses module Data.forecast_dataset module Data.Synthetic.generate_synthetic_data module

Data.Synthetic.synthetic_data_util module data_split() (Data.forecast_dataset.WeatherDataset method) DataSaver (class in Results.save_results)

E

early_stop() (Training.train_utils.EarlyStopper method) EarlyStopper (class in Training.train_utils)

EGP (class in Models.GP) end_run() (Results.save_results.DataSaver method) era5_name_to_kis_name() (Data.forecast_dataset.GridStationDataset method)

F

FNN (class in Models.FNN) forward() (Data.data_losses.Full_MSELoss method) (Data.data_losses.GradientSmoothnessLoss method) (Data.data_losses.Mean_MSELoss method) (Data.data_losses.NMLLLoss method) (Data.data_losses.VariationalELBOLoss method) (Models.AFNO.AFNONet method) (Models.FNN.FNN method) (Models.GP.EGP method) (Models.GP.GPWrapper method) (Models.GP.VariationalGP method) (Models.GST.CGST method) (Models.GST.MGST method) (Models.neural_models.ACW method)

forward_features() (Models.AFNO.AFNONet method) Full_MSELoss (class in Data.data_losses)

G

get_base_dir() (Results.save_results.DataSaver method) get_data_config() (Results.parser.ConfigManager method) get_dataset() (in module Results.extract_from_configuration) get_domain_shape() (Data.forecast_dataset.GridStationDataset method) (Data.forecast_dataset.PastNDaysForecastDataset method) (Data.forecast_dataset.WeatherDataset method) get_in_channels() (Data.forecast_dataset.GridStationDataset method) (Data.forecast_dataset.PastNDaysForecastDataset method) (Data.forecast_dataset.WeatherDataset method) get_kis_obs() (Data.forecast_dataset.GridStationDataset method) get_model_config() (Results.parser.ConfigManager method) get_num_train_data() (Data.forecast_dataset.GridStationDataset method) get_output_config() (Results.parser.ConfigManager method) get_prob_model() (Models.neural_models.PNN method) get_run_dir() (Results.save_results.DataSaver method) get_test_idx() (Data.forecast_dataset.GridStationDataset method)

get_test_loader() (Data.forecast_dataset.GridStationDataset method) (Data.forecast_dataset.PastNDaysForecastDataset method) (Data.forecast_dataset.WeatherDataset method) get_test_xy() (Data.forecast_dataset.PastNDaysForecastDataset method) (Data.forecast_dataset.WeatherDataset method) get_timestamp() (Data.forecast_dataset.GridStationDataset method) get_train_idx() (Data.forecast_dataset.GridStationDataset method) get_train_loader() (Data.forecast_dataset.GridStationDataset method) (Data.forecast_dataset.WeatherDataset method) get_train_xy() (Data.forecast_dataset.WeatherDataset method) get_training_components() (in module Results.extract_from_configuration) get_training_config() (Results.parser.ConfigManager method) get_vmins_and_vmaxs() (in module Data.Synthetic.synthetic_data_util) GPWrapper (class in Models.GP) GradientSmoothnessLoss (class in Data.data_losses) GridStationDataset (class in Data.forecast_dataset)

I

initialize_normalization_stats() (Models.GP.GPWrapper method)

integrand() (in module Data.Synthetic.synthetic_data_util) interpolate_era5() (Data.forecast_dataset.GridStationDataset method)

L

load_config() (Results.parser.ConfigManager method)

load_state_dict() (Models.VGP_optimizer.VGPOptimizer method) log_metrics() (Results.save_results.DataSaver method)

M

Mean_MSELoss (class in Data.data_losses) MGST (class in Models.GST) Models.AFNO module Models.FNN module Models.GP module Models.GST module Models.neural_models module Models.VGP_optimizer module module Data.data_losses Data.forecast_dataset Data.Synthetic.generate_synthetic_data Data.Synthetic.synthetic_data_util Models.AFNO Models.FNN Models.GP Models.GST Models.neural_models Models.VGP_optimizer Results.extract_from_configuration Results.parser Results.run_configuration Results.save_results Training.train_utils

N

NMLLLoss (class in Data.data_losses)

no_weight_decay() (Models.AFNO.AFNONet method)

P

PastNDaysForecastDataset (class in Data.forecast_dataset) plot_contours() (in module Data.Synthetic.synthetic_data_util)

PNN (class in Models.neural_models) precompute_kis_obs() (Data.forecast_dataset.GridStationDataset method) precompute_outputs() (Data.forecast_dataset.GridStationDataset method)

R

Results.extract_from_configuration module Results.parser module Results.run_configuration module

Results.save_results module run_configuration() (in module Results.run_configuration) run_folder() (in module Results.run_configuration)

S

sample() (Models.neural_models.BNN method) save_config() (Results.parser.ConfigManager method) (Results.save_results.DataSaver method) save_metrics() (Results.save_results.DataSaver method) save_model() (Results.save_results.DataSaver method) shuffle_train_indices() (Data.forecast_dataset.GridStationDataset method)

sigma_t() (in module Data.Synthetic.synthetic_data_util) sigma_w() (in module Data.Synthetic.synthetic_data_util) sigma_x() (in module Data.Synthetic.synthetic_data_util) state_dict() (Models.VGP_optimizer.VGPOptimizer method) step() (Models.VGP_optimizer.VGPOptimizer method) str_to_class() (in module Results.extract_from_configuration)

T

temp() (in module Data.Synthetic.synthetic_data_util) temp_h() (in module Data.Synthetic.synthetic_data_util) temp_i() (in module Data.Synthetic.synthetic_data_util) temp_p() (in module Data.Synthetic.synthetic_data_util) to() (Data.forecast_dataset.GridStationDataset method) (Data.forecast_dataset.PastNDaysForecastDataset method)

train_and_evaluate_model() (in module Training.train_utils) train_model() (in module Training.train_utils) train_one_epoch() (in module Training.train_utils) Training.train_utils module

V

val_one_epoch() (in module Training.train_utils) VariationalELBOLoss (class in Data.data_losses)

VariationalGP (class in Models.GP) VGPOptimizer (class in Models.VGP_optimizer)

W

wandb_summary() (Results.save_results.DataSaver method)

WeatherDataset (class in Data.forecast_dataset)

Z

zero_grad() (Models.VGP_optimizer.VGPOptimizer method)