├── .github └── workflows │ ├── build-book.yaml │ ├── deploy-book.yaml │ └── publish-website.yml ├── .gitignore ├── LICENSE ├── README.md ├── chaosbench ├── config.py ├── configs │ ├── ed_s2s.yaml │ ├── fno_s2s.yaml │ ├── resnet_ensemble_s2s.yaml │ ├── resnet_s2s.yaml │ ├── segformer_s2s.yaml │ ├── unet_ensemble_s2s.yaml │ ├── unet_s2s.yaml │ └── vae_s2s.yaml ├── criterion.py ├── dataset.py ├── models │ ├── ae.py │ ├── cnn.py │ ├── fno.py │ ├── mlp.py │ ├── model.py │ └── vit.py └── utils.py ├── data └── .gitkeep ├── docs ├── 3d_climax_Task_1.pdf ├── 3d_climax_Task_2.pdf ├── 3d_ed_Task_1.pdf ├── 3d_ed_Task_2.pdf ├── 3d_fno_Task_1.pdf ├── 3d_fno_Task_2.pdf ├── 3d_fourcastnetv2_sota.pdf ├── 3d_graphcast_sota.pdf ├── 3d_panguweather_sota.pdf ├── 3d_resnet_Task_1.pdf ├── 3d_resnet_Task_2.pdf ├── 3d_unet_Task_1.pdf ├── 3d_unet_Task_2.pdf ├── 3d_unet_s2s_Task_1.pdf ├── 3d_unet_s2s_Task_2.pdf ├── all_rmse_sota.png ├── autoreg_acc_task1.pdf ├── autoreg_acc_task2.pdf ├── autoreg_rmse_task1.pdf ├── autoreg_rmse_task2.pdf ├── autoreg_sdiv_task1.pdf ├── autoreg_sdiv_task2.pdf ├── autoreg_ssim_task1.pdf ├── autoreg_ssim_task2.pdf ├── bc_acc.pdf ├── bc_rmse.pdf ├── bc_ssim.pdf ├── center_acc.pdf ├── center_bias.pdf ├── center_ens_acc.pdf ├── center_ens_bias.pdf ├── center_ens_rmse.pdf ├── center_ens_sdiv.pdf ├── center_ens_ssim.pdf ├── center_probs_crps.pdf ├── center_probs_crpss.pdf ├── center_probs_spread.pdf ├── center_probs_ssr.pdf ├── center_ratio_acc.pdf ├── center_ratio_bias.pdf ├── center_ratio_rmse.pdf ├── center_ratio_sdiv.pdf ├── center_ratio_ssim.pdf ├── center_rmse.pdf ├── center_sdiv.pdf ├── center_ssim.pdf ├── energy_spectra_3d.pdf ├── ml_probs_crps.pdf ├── ml_probs_crpss.pdf ├── ml_probs_spread.pdf ├── ml_probs_ssr.pdf ├── ml_ratio_acc.pdf ├── ml_ratio_bias.pdf ├── ml_ratio_rmse.pdf ├── ml_ratio_sdiv.pdf ├── ml_ratio_ssim.pdf ├── preds_climax_q700_direct_Task 1.pdf ├── preds_climax_q700_direct_Task 2.pdf ├── preds_climax_q700_direct_Task1.png ├── preds_climax_t850_direct_Task 1.pdf ├── preds_climax_t850_direct_Task 2.pdf ├── preds_climax_z500_direct_Task 1.pdf ├── preds_climax_z500_direct_Task 2.pdf ├── preds_unet_s2s_q700_S=1_Task 1.pdf ├── preds_unet_s2s_q700_S=1_Task 2.pdf ├── preds_unet_s2s_q700_S=5_Task 1.pdf ├── preds_unet_s2s_q700_S=5_Task 2.pdf ├── preds_unet_s2s_q700_direct_Task 1.pdf ├── preds_unet_s2s_q700_direct_Task 2.pdf ├── preds_unet_s2s_t850_S=1_Task 1.pdf ├── preds_unet_s2s_t850_S=1_Task 2.pdf ├── preds_unet_s2s_t850_S=5_Task 1.pdf ├── preds_unet_s2s_t850_S=5_Task 2.pdf ├── preds_unet_s2s_t850_direct_Task 1.pdf ├── preds_unet_s2s_t850_direct_Task 2.pdf ├── preds_unet_s2s_z500_S=1_Task 1.pdf ├── preds_unet_s2s_z500_S=1_Task 2.pdf ├── preds_unet_s2s_z500_S=5_Task 1.pdf ├── preds_unet_s2s_z500_S=5_Task 2.pdf ├── preds_unet_s2s_z500_direct_Task 1.pdf ├── preds_unet_s2s_z500_direct_Task 2.pdf ├── scheme │ ├── chaosbench_scheme-physics-model.jpg │ └── chaosbench_scheme-scheme.jpg ├── signal_loss.pdf ├── sota_acc.pdf ├── sota_bias.pdf ├── sota_rmse.pdf ├── sota_sdiv.pdf ├── sota_ssim.pdf ├── specdiv_climax_direct_Task 1.pdf ├── specdiv_climax_direct_Task 2.pdf ├── specdiv_ed_Task 1.pdf ├── specdiv_ed_Task 2.pdf ├── specdiv_fno_Task 1.pdf ├── specdiv_fno_Task 2.pdf ├── specdiv_fourcastnetv2_sota.pdf ├── specdiv_graphcast_sota.pdf ├── specdiv_panguweather_sota.pdf ├── specdiv_resnet_Task 1.pdf ├── specdiv_resnet_Task 2.pdf ├── specdiv_unet_Task 1.pdf ├── specdiv_unet_Task 2.pdf ├── specdiv_unet_s2s_direct_Task 1.pdf ├── specdiv_unet_s2s_direct_Task 2.pdf ├── time_bias.pdf ├── time_rmse.pdf ├── time_sdiv.pdf └── time_ssim.pdf ├── eval_direct.py ├── eval_ensemble.py ├── eval_iter.py ├── eval_sota.py ├── logs └── .gitkeep ├── notebooks ├── 01a_s2s_data_exploration.ipynb ├── 02a_s2s_modeling.ipynb ├── 03a_s2s_train.ipynb ├── 04a_s2s_eval_iter.ipynb └── 04b_s2s_eval_direct.ipynb ├── requirements.txt ├── scripts ├── compute_climatology.py ├── config.py ├── generate_interactive_html.py ├── process_atmos.py ├── process_center.py ├── process_land.py ├── process_ocean.py └── process_sota.py ├── train.py └── website ├── _config.yml ├── _toc.yml ├── baseline.md ├── dataset.md ├── environment.yml ├── evaluation.md ├── html ├── control.html └── ensemble.html ├── leaderboard.md ├── metrics.md ├── motivation.md ├── quickstart.md ├── task.md └── training.md /.github/workflows/build-book.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/.github/workflows/build-book.yaml -------------------------------------------------------------------------------- /.github/workflows/deploy-book.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/.github/workflows/deploy-book.yaml -------------------------------------------------------------------------------- /.github/workflows/publish-website.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/.github/workflows/publish-website.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/README.md -------------------------------------------------------------------------------- /chaosbench/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/chaosbench/config.py -------------------------------------------------------------------------------- /chaosbench/configs/ed_s2s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/chaosbench/configs/ed_s2s.yaml -------------------------------------------------------------------------------- /chaosbench/configs/fno_s2s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/chaosbench/configs/fno_s2s.yaml -------------------------------------------------------------------------------- /chaosbench/configs/resnet_ensemble_s2s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/chaosbench/configs/resnet_ensemble_s2s.yaml -------------------------------------------------------------------------------- /chaosbench/configs/resnet_s2s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/chaosbench/configs/resnet_s2s.yaml -------------------------------------------------------------------------------- /chaosbench/configs/segformer_s2s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/chaosbench/configs/segformer_s2s.yaml -------------------------------------------------------------------------------- /chaosbench/configs/unet_ensemble_s2s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/chaosbench/configs/unet_ensemble_s2s.yaml -------------------------------------------------------------------------------- /chaosbench/configs/unet_s2s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/chaosbench/configs/unet_s2s.yaml -------------------------------------------------------------------------------- /chaosbench/configs/vae_s2s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/chaosbench/configs/vae_s2s.yaml -------------------------------------------------------------------------------- /chaosbench/criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/chaosbench/criterion.py -------------------------------------------------------------------------------- /chaosbench/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/chaosbench/dataset.py -------------------------------------------------------------------------------- /chaosbench/models/ae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/chaosbench/models/ae.py -------------------------------------------------------------------------------- /chaosbench/models/cnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/chaosbench/models/cnn.py -------------------------------------------------------------------------------- /chaosbench/models/fno.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/chaosbench/models/fno.py -------------------------------------------------------------------------------- /chaosbench/models/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/chaosbench/models/mlp.py -------------------------------------------------------------------------------- /chaosbench/models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/chaosbench/models/model.py -------------------------------------------------------------------------------- /chaosbench/models/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/chaosbench/models/vit.py -------------------------------------------------------------------------------- /chaosbench/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/chaosbench/utils.py -------------------------------------------------------------------------------- /data/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /docs/3d_climax_Task_1.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/3d_climax_Task_1.pdf -------------------------------------------------------------------------------- /docs/3d_climax_Task_2.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/3d_climax_Task_2.pdf -------------------------------------------------------------------------------- /docs/3d_ed_Task_1.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/3d_ed_Task_1.pdf -------------------------------------------------------------------------------- /docs/3d_ed_Task_2.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/3d_ed_Task_2.pdf -------------------------------------------------------------------------------- /docs/3d_fno_Task_1.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/3d_fno_Task_1.pdf -------------------------------------------------------------------------------- /docs/3d_fno_Task_2.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/3d_fno_Task_2.pdf -------------------------------------------------------------------------------- /docs/3d_fourcastnetv2_sota.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/3d_fourcastnetv2_sota.pdf -------------------------------------------------------------------------------- /docs/3d_graphcast_sota.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/3d_graphcast_sota.pdf -------------------------------------------------------------------------------- /docs/3d_panguweather_sota.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/3d_panguweather_sota.pdf -------------------------------------------------------------------------------- /docs/3d_resnet_Task_1.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/3d_resnet_Task_1.pdf -------------------------------------------------------------------------------- /docs/3d_resnet_Task_2.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/3d_resnet_Task_2.pdf -------------------------------------------------------------------------------- /docs/3d_unet_Task_1.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/3d_unet_Task_1.pdf -------------------------------------------------------------------------------- /docs/3d_unet_Task_2.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/3d_unet_Task_2.pdf -------------------------------------------------------------------------------- /docs/3d_unet_s2s_Task_1.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/3d_unet_s2s_Task_1.pdf -------------------------------------------------------------------------------- /docs/3d_unet_s2s_Task_2.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/3d_unet_s2s_Task_2.pdf -------------------------------------------------------------------------------- /docs/all_rmse_sota.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/all_rmse_sota.png -------------------------------------------------------------------------------- /docs/autoreg_acc_task1.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/autoreg_acc_task1.pdf -------------------------------------------------------------------------------- /docs/autoreg_acc_task2.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/autoreg_acc_task2.pdf -------------------------------------------------------------------------------- /docs/autoreg_rmse_task1.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/autoreg_rmse_task1.pdf -------------------------------------------------------------------------------- /docs/autoreg_rmse_task2.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/autoreg_rmse_task2.pdf -------------------------------------------------------------------------------- /docs/autoreg_sdiv_task1.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/autoreg_sdiv_task1.pdf -------------------------------------------------------------------------------- /docs/autoreg_sdiv_task2.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/autoreg_sdiv_task2.pdf -------------------------------------------------------------------------------- /docs/autoreg_ssim_task1.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/autoreg_ssim_task1.pdf -------------------------------------------------------------------------------- /docs/autoreg_ssim_task2.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/autoreg_ssim_task2.pdf -------------------------------------------------------------------------------- /docs/bc_acc.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/bc_acc.pdf -------------------------------------------------------------------------------- /docs/bc_rmse.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/bc_rmse.pdf -------------------------------------------------------------------------------- /docs/bc_ssim.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/bc_ssim.pdf -------------------------------------------------------------------------------- /docs/center_acc.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/center_acc.pdf -------------------------------------------------------------------------------- /docs/center_bias.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/center_bias.pdf -------------------------------------------------------------------------------- /docs/center_ens_acc.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/center_ens_acc.pdf -------------------------------------------------------------------------------- /docs/center_ens_bias.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/center_ens_bias.pdf -------------------------------------------------------------------------------- /docs/center_ens_rmse.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/center_ens_rmse.pdf -------------------------------------------------------------------------------- /docs/center_ens_sdiv.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/center_ens_sdiv.pdf -------------------------------------------------------------------------------- /docs/center_ens_ssim.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/center_ens_ssim.pdf -------------------------------------------------------------------------------- /docs/center_probs_crps.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/center_probs_crps.pdf -------------------------------------------------------------------------------- /docs/center_probs_crpss.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/center_probs_crpss.pdf -------------------------------------------------------------------------------- /docs/center_probs_spread.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/center_probs_spread.pdf -------------------------------------------------------------------------------- /docs/center_probs_ssr.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/center_probs_ssr.pdf -------------------------------------------------------------------------------- /docs/center_ratio_acc.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/center_ratio_acc.pdf -------------------------------------------------------------------------------- /docs/center_ratio_bias.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/center_ratio_bias.pdf -------------------------------------------------------------------------------- /docs/center_ratio_rmse.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/center_ratio_rmse.pdf -------------------------------------------------------------------------------- /docs/center_ratio_sdiv.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/center_ratio_sdiv.pdf -------------------------------------------------------------------------------- /docs/center_ratio_ssim.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/center_ratio_ssim.pdf -------------------------------------------------------------------------------- /docs/center_rmse.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/center_rmse.pdf -------------------------------------------------------------------------------- /docs/center_sdiv.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/center_sdiv.pdf -------------------------------------------------------------------------------- /docs/center_ssim.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/center_ssim.pdf -------------------------------------------------------------------------------- /docs/energy_spectra_3d.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/energy_spectra_3d.pdf -------------------------------------------------------------------------------- /docs/ml_probs_crps.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/ml_probs_crps.pdf -------------------------------------------------------------------------------- /docs/ml_probs_crpss.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/ml_probs_crpss.pdf -------------------------------------------------------------------------------- /docs/ml_probs_spread.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/ml_probs_spread.pdf -------------------------------------------------------------------------------- /docs/ml_probs_ssr.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/ml_probs_ssr.pdf -------------------------------------------------------------------------------- /docs/ml_ratio_acc.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/ml_ratio_acc.pdf -------------------------------------------------------------------------------- /docs/ml_ratio_bias.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/ml_ratio_bias.pdf -------------------------------------------------------------------------------- /docs/ml_ratio_rmse.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/ml_ratio_rmse.pdf -------------------------------------------------------------------------------- /docs/ml_ratio_sdiv.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/ml_ratio_sdiv.pdf -------------------------------------------------------------------------------- /docs/ml_ratio_ssim.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/ml_ratio_ssim.pdf -------------------------------------------------------------------------------- /docs/preds_climax_q700_direct_Task 1.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/preds_climax_q700_direct_Task 1.pdf -------------------------------------------------------------------------------- /docs/preds_climax_q700_direct_Task 2.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/preds_climax_q700_direct_Task 2.pdf -------------------------------------------------------------------------------- /docs/preds_climax_q700_direct_Task1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/preds_climax_q700_direct_Task1.png -------------------------------------------------------------------------------- /docs/preds_climax_t850_direct_Task 1.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/preds_climax_t850_direct_Task 1.pdf -------------------------------------------------------------------------------- /docs/preds_climax_t850_direct_Task 2.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/preds_climax_t850_direct_Task 2.pdf -------------------------------------------------------------------------------- /docs/preds_climax_z500_direct_Task 1.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/preds_climax_z500_direct_Task 1.pdf -------------------------------------------------------------------------------- /docs/preds_climax_z500_direct_Task 2.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/preds_climax_z500_direct_Task 2.pdf -------------------------------------------------------------------------------- /docs/preds_unet_s2s_q700_S=1_Task 1.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/preds_unet_s2s_q700_S=1_Task 1.pdf -------------------------------------------------------------------------------- /docs/preds_unet_s2s_q700_S=1_Task 2.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/preds_unet_s2s_q700_S=1_Task 2.pdf -------------------------------------------------------------------------------- /docs/preds_unet_s2s_q700_S=5_Task 1.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/preds_unet_s2s_q700_S=5_Task 1.pdf -------------------------------------------------------------------------------- /docs/preds_unet_s2s_q700_S=5_Task 2.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/preds_unet_s2s_q700_S=5_Task 2.pdf -------------------------------------------------------------------------------- /docs/preds_unet_s2s_q700_direct_Task 1.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/preds_unet_s2s_q700_direct_Task 1.pdf -------------------------------------------------------------------------------- /docs/preds_unet_s2s_q700_direct_Task 2.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/preds_unet_s2s_q700_direct_Task 2.pdf -------------------------------------------------------------------------------- /docs/preds_unet_s2s_t850_S=1_Task 1.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/preds_unet_s2s_t850_S=1_Task 1.pdf -------------------------------------------------------------------------------- /docs/preds_unet_s2s_t850_S=1_Task 2.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/preds_unet_s2s_t850_S=1_Task 2.pdf -------------------------------------------------------------------------------- /docs/preds_unet_s2s_t850_S=5_Task 1.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/preds_unet_s2s_t850_S=5_Task 1.pdf -------------------------------------------------------------------------------- /docs/preds_unet_s2s_t850_S=5_Task 2.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/preds_unet_s2s_t850_S=5_Task 2.pdf -------------------------------------------------------------------------------- /docs/preds_unet_s2s_t850_direct_Task 1.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/preds_unet_s2s_t850_direct_Task 1.pdf -------------------------------------------------------------------------------- /docs/preds_unet_s2s_t850_direct_Task 2.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/preds_unet_s2s_t850_direct_Task 2.pdf -------------------------------------------------------------------------------- /docs/preds_unet_s2s_z500_S=1_Task 1.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/preds_unet_s2s_z500_S=1_Task 1.pdf -------------------------------------------------------------------------------- /docs/preds_unet_s2s_z500_S=1_Task 2.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/preds_unet_s2s_z500_S=1_Task 2.pdf -------------------------------------------------------------------------------- /docs/preds_unet_s2s_z500_S=5_Task 1.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/preds_unet_s2s_z500_S=5_Task 1.pdf -------------------------------------------------------------------------------- /docs/preds_unet_s2s_z500_S=5_Task 2.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/preds_unet_s2s_z500_S=5_Task 2.pdf -------------------------------------------------------------------------------- /docs/preds_unet_s2s_z500_direct_Task 1.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/preds_unet_s2s_z500_direct_Task 1.pdf -------------------------------------------------------------------------------- /docs/preds_unet_s2s_z500_direct_Task 2.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/preds_unet_s2s_z500_direct_Task 2.pdf -------------------------------------------------------------------------------- /docs/scheme/chaosbench_scheme-physics-model.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/scheme/chaosbench_scheme-physics-model.jpg -------------------------------------------------------------------------------- /docs/scheme/chaosbench_scheme-scheme.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/scheme/chaosbench_scheme-scheme.jpg -------------------------------------------------------------------------------- /docs/signal_loss.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/signal_loss.pdf -------------------------------------------------------------------------------- /docs/sota_acc.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/sota_acc.pdf -------------------------------------------------------------------------------- /docs/sota_bias.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/sota_bias.pdf -------------------------------------------------------------------------------- /docs/sota_rmse.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/sota_rmse.pdf -------------------------------------------------------------------------------- /docs/sota_sdiv.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/sota_sdiv.pdf -------------------------------------------------------------------------------- /docs/sota_ssim.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/sota_ssim.pdf -------------------------------------------------------------------------------- /docs/specdiv_climax_direct_Task 1.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/specdiv_climax_direct_Task 1.pdf -------------------------------------------------------------------------------- /docs/specdiv_climax_direct_Task 2.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/specdiv_climax_direct_Task 2.pdf -------------------------------------------------------------------------------- /docs/specdiv_ed_Task 1.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/specdiv_ed_Task 1.pdf -------------------------------------------------------------------------------- /docs/specdiv_ed_Task 2.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/specdiv_ed_Task 2.pdf -------------------------------------------------------------------------------- /docs/specdiv_fno_Task 1.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/specdiv_fno_Task 1.pdf -------------------------------------------------------------------------------- /docs/specdiv_fno_Task 2.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/specdiv_fno_Task 2.pdf -------------------------------------------------------------------------------- /docs/specdiv_fourcastnetv2_sota.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/specdiv_fourcastnetv2_sota.pdf -------------------------------------------------------------------------------- /docs/specdiv_graphcast_sota.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/specdiv_graphcast_sota.pdf -------------------------------------------------------------------------------- /docs/specdiv_panguweather_sota.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/specdiv_panguweather_sota.pdf -------------------------------------------------------------------------------- /docs/specdiv_resnet_Task 1.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/specdiv_resnet_Task 1.pdf -------------------------------------------------------------------------------- /docs/specdiv_resnet_Task 2.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/specdiv_resnet_Task 2.pdf -------------------------------------------------------------------------------- /docs/specdiv_unet_Task 1.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/specdiv_unet_Task 1.pdf -------------------------------------------------------------------------------- /docs/specdiv_unet_Task 2.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/specdiv_unet_Task 2.pdf -------------------------------------------------------------------------------- /docs/specdiv_unet_s2s_direct_Task 1.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/specdiv_unet_s2s_direct_Task 1.pdf -------------------------------------------------------------------------------- /docs/specdiv_unet_s2s_direct_Task 2.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/specdiv_unet_s2s_direct_Task 2.pdf -------------------------------------------------------------------------------- /docs/time_bias.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/time_bias.pdf -------------------------------------------------------------------------------- /docs/time_rmse.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/time_rmse.pdf -------------------------------------------------------------------------------- /docs/time_sdiv.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/time_sdiv.pdf -------------------------------------------------------------------------------- /docs/time_ssim.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/docs/time_ssim.pdf -------------------------------------------------------------------------------- /eval_direct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/eval_direct.py -------------------------------------------------------------------------------- /eval_ensemble.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/eval_ensemble.py -------------------------------------------------------------------------------- /eval_iter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/eval_iter.py -------------------------------------------------------------------------------- /eval_sota.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/eval_sota.py -------------------------------------------------------------------------------- /logs/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /notebooks/01a_s2s_data_exploration.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/notebooks/01a_s2s_data_exploration.ipynb -------------------------------------------------------------------------------- /notebooks/02a_s2s_modeling.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/notebooks/02a_s2s_modeling.ipynb -------------------------------------------------------------------------------- /notebooks/03a_s2s_train.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/notebooks/03a_s2s_train.ipynb -------------------------------------------------------------------------------- /notebooks/04a_s2s_eval_iter.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/notebooks/04a_s2s_eval_iter.ipynb -------------------------------------------------------------------------------- /notebooks/04b_s2s_eval_direct.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/notebooks/04b_s2s_eval_direct.ipynb -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/compute_climatology.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/scripts/compute_climatology.py -------------------------------------------------------------------------------- /scripts/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/scripts/config.py -------------------------------------------------------------------------------- /scripts/generate_interactive_html.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/scripts/generate_interactive_html.py -------------------------------------------------------------------------------- /scripts/process_atmos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/scripts/process_atmos.py -------------------------------------------------------------------------------- /scripts/process_center.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/scripts/process_center.py -------------------------------------------------------------------------------- /scripts/process_land.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/scripts/process_land.py -------------------------------------------------------------------------------- /scripts/process_ocean.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/scripts/process_ocean.py -------------------------------------------------------------------------------- /scripts/process_sota.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/scripts/process_sota.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/train.py -------------------------------------------------------------------------------- /website/_config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/website/_config.yml -------------------------------------------------------------------------------- /website/_toc.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/website/_toc.yml -------------------------------------------------------------------------------- /website/baseline.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/website/baseline.md -------------------------------------------------------------------------------- /website/dataset.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/website/dataset.md -------------------------------------------------------------------------------- /website/environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/website/environment.yml -------------------------------------------------------------------------------- /website/evaluation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/website/evaluation.md -------------------------------------------------------------------------------- /website/html/control.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/website/html/control.html -------------------------------------------------------------------------------- /website/html/ensemble.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/website/html/ensemble.html -------------------------------------------------------------------------------- /website/leaderboard.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/website/leaderboard.md -------------------------------------------------------------------------------- /website/metrics.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/website/metrics.md -------------------------------------------------------------------------------- /website/motivation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/website/motivation.md -------------------------------------------------------------------------------- /website/quickstart.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/website/quickstart.md -------------------------------------------------------------------------------- /website/task.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/website/task.md -------------------------------------------------------------------------------- /website/training.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leap-stc/ChaosBench/HEAD/website/training.md --------------------------------------------------------------------------------