├── .github └── workflows │ ├── inactive-issues.yml │ └── main.yml ├── .gitignore ├── LICENSE ├── MANIFEST.in ├── README.md ├── climatereconstructionai ├── __init__.py ├── config.py ├── evaluate.py ├── loss │ ├── feature_loss.py │ ├── get_loss.py │ ├── hole_loss.py │ ├── total_variation_loss.py │ ├── utils.py │ └── valid_loss.py ├── metrics │ └── get_metrics.py ├── model │ ├── attention_module.py │ ├── bounds_scaler.py │ ├── conv_configs.py │ ├── conv_lstm_module.py │ ├── encoder_decoder.py │ ├── net.py │ ├── partial_conv_module.py │ └── traj_gru_module.py ├── static │ └── dataset_format.json ├── train.py └── utils │ ├── early_stopping.py │ ├── evaluation.py │ ├── featurizer.py │ ├── io.py │ ├── masked_batchnorm.py │ ├── netcdfchecker.py │ ├── netcdfloader.py │ ├── normalizer.py │ ├── plotdata.py │ ├── profiler.py │ ├── twriter.py │ ├── visualization.py │ └── weights.py ├── data ├── 20crtasgn72.pth ├── hadcrut_missmask_1.nc ├── test │ └── tas_hadcrut_187709_189308.nc ├── train │ ├── 20cr-1ens.nc │ └── 20cr.json └── val │ ├── 20cr-1ens.nc │ └── 20cr.json ├── demo ├── README.md ├── demo_args.txt ├── images │ └── demo_combined.1_0.png └── outputs │ └── .gitignore ├── environment-cuda.yml ├── environment.yml ├── pyproject.toml ├── pytest.ini ├── requirements.txt ├── setup.py └── tests ├── .gitignore ├── in ├── evaluation │ └── minimum-1.inp └── training │ ├── attention-channel-memory.inp │ ├── attention.inp │ ├── channel-infusion.inp │ ├── channel-memory.inp │ ├── conv-lstm.inp │ ├── json-input.inp │ ├── minimum-1.inp │ ├── minimum-2.inp │ ├── target-channel-memory.inp │ └── traj-gru.inp ├── ref ├── attention-channel-memory.inp.pth ├── attention.inp.pth ├── channel-infusion.inp.pth ├── channel-memory.inp.pth ├── conv-lstm.inp.pth ├── json-input.inp.pth ├── minimum-1.inp.pth ├── minimum-1_infilled.nc ├── minimum-2.inp.pth ├── target-channel-memory.inp.pth └── traj-gru.inp.pth ├── test_evaluation.py └── test_training.py /.github/workflows/inactive-issues.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/.github/workflows/inactive-issues.yml -------------------------------------------------------------------------------- /.github/workflows/main.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/.github/workflows/main.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- 1 | include climatereconstructionai/static/*.json -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/README.md -------------------------------------------------------------------------------- /climatereconstructionai/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/climatereconstructionai/__init__.py -------------------------------------------------------------------------------- /climatereconstructionai/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/climatereconstructionai/config.py -------------------------------------------------------------------------------- /climatereconstructionai/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/climatereconstructionai/evaluate.py -------------------------------------------------------------------------------- /climatereconstructionai/loss/feature_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/climatereconstructionai/loss/feature_loss.py -------------------------------------------------------------------------------- /climatereconstructionai/loss/get_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/climatereconstructionai/loss/get_loss.py -------------------------------------------------------------------------------- /climatereconstructionai/loss/hole_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/climatereconstructionai/loss/hole_loss.py -------------------------------------------------------------------------------- /climatereconstructionai/loss/total_variation_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/climatereconstructionai/loss/total_variation_loss.py -------------------------------------------------------------------------------- /climatereconstructionai/loss/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/climatereconstructionai/loss/utils.py -------------------------------------------------------------------------------- /climatereconstructionai/loss/valid_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/climatereconstructionai/loss/valid_loss.py -------------------------------------------------------------------------------- /climatereconstructionai/metrics/get_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/climatereconstructionai/metrics/get_metrics.py -------------------------------------------------------------------------------- /climatereconstructionai/model/attention_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/climatereconstructionai/model/attention_module.py -------------------------------------------------------------------------------- /climatereconstructionai/model/bounds_scaler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/climatereconstructionai/model/bounds_scaler.py -------------------------------------------------------------------------------- /climatereconstructionai/model/conv_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/climatereconstructionai/model/conv_configs.py -------------------------------------------------------------------------------- /climatereconstructionai/model/conv_lstm_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/climatereconstructionai/model/conv_lstm_module.py -------------------------------------------------------------------------------- /climatereconstructionai/model/encoder_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/climatereconstructionai/model/encoder_decoder.py -------------------------------------------------------------------------------- /climatereconstructionai/model/net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/climatereconstructionai/model/net.py -------------------------------------------------------------------------------- /climatereconstructionai/model/partial_conv_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/climatereconstructionai/model/partial_conv_module.py -------------------------------------------------------------------------------- /climatereconstructionai/model/traj_gru_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/climatereconstructionai/model/traj_gru_module.py -------------------------------------------------------------------------------- /climatereconstructionai/static/dataset_format.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/climatereconstructionai/static/dataset_format.json -------------------------------------------------------------------------------- /climatereconstructionai/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/climatereconstructionai/train.py -------------------------------------------------------------------------------- /climatereconstructionai/utils/early_stopping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/climatereconstructionai/utils/early_stopping.py -------------------------------------------------------------------------------- /climatereconstructionai/utils/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/climatereconstructionai/utils/evaluation.py -------------------------------------------------------------------------------- /climatereconstructionai/utils/featurizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/climatereconstructionai/utils/featurizer.py -------------------------------------------------------------------------------- /climatereconstructionai/utils/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/climatereconstructionai/utils/io.py -------------------------------------------------------------------------------- /climatereconstructionai/utils/masked_batchnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/climatereconstructionai/utils/masked_batchnorm.py -------------------------------------------------------------------------------- /climatereconstructionai/utils/netcdfchecker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/climatereconstructionai/utils/netcdfchecker.py -------------------------------------------------------------------------------- /climatereconstructionai/utils/netcdfloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/climatereconstructionai/utils/netcdfloader.py -------------------------------------------------------------------------------- /climatereconstructionai/utils/normalizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/climatereconstructionai/utils/normalizer.py -------------------------------------------------------------------------------- /climatereconstructionai/utils/plotdata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/climatereconstructionai/utils/plotdata.py -------------------------------------------------------------------------------- /climatereconstructionai/utils/profiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/climatereconstructionai/utils/profiler.py -------------------------------------------------------------------------------- /climatereconstructionai/utils/twriter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/climatereconstructionai/utils/twriter.py -------------------------------------------------------------------------------- /climatereconstructionai/utils/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/climatereconstructionai/utils/visualization.py -------------------------------------------------------------------------------- /climatereconstructionai/utils/weights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/climatereconstructionai/utils/weights.py -------------------------------------------------------------------------------- /data/20crtasgn72.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/data/20crtasgn72.pth -------------------------------------------------------------------------------- /data/hadcrut_missmask_1.nc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/data/hadcrut_missmask_1.nc -------------------------------------------------------------------------------- /data/test/tas_hadcrut_187709_189308.nc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/data/test/tas_hadcrut_187709_189308.nc -------------------------------------------------------------------------------- /data/train/20cr-1ens.nc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/data/train/20cr-1ens.nc -------------------------------------------------------------------------------- /data/train/20cr.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/data/train/20cr.json -------------------------------------------------------------------------------- /data/val/20cr-1ens.nc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/data/val/20cr-1ens.nc -------------------------------------------------------------------------------- /data/val/20cr.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/data/val/20cr.json -------------------------------------------------------------------------------- /demo/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/demo/README.md -------------------------------------------------------------------------------- /demo/demo_args.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/demo/demo_args.txt -------------------------------------------------------------------------------- /demo/images/demo_combined.1_0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/demo/images/demo_combined.1_0.png -------------------------------------------------------------------------------- /demo/outputs/.gitignore: -------------------------------------------------------------------------------- 1 | *.nc 2 | *.png 3 | -------------------------------------------------------------------------------- /environment-cuda.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/environment-cuda.yml -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/environment.yml -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/pyproject.toml -------------------------------------------------------------------------------- /pytest.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/pytest.ini -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/setup.py -------------------------------------------------------------------------------- /tests/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/tests/.gitignore -------------------------------------------------------------------------------- /tests/in/evaluation/minimum-1.inp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/tests/in/evaluation/minimum-1.inp -------------------------------------------------------------------------------- /tests/in/training/attention-channel-memory.inp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/tests/in/training/attention-channel-memory.inp -------------------------------------------------------------------------------- /tests/in/training/attention.inp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/tests/in/training/attention.inp -------------------------------------------------------------------------------- /tests/in/training/channel-infusion.inp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/tests/in/training/channel-infusion.inp -------------------------------------------------------------------------------- /tests/in/training/channel-memory.inp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/tests/in/training/channel-memory.inp -------------------------------------------------------------------------------- /tests/in/training/conv-lstm.inp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/tests/in/training/conv-lstm.inp -------------------------------------------------------------------------------- /tests/in/training/json-input.inp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/tests/in/training/json-input.inp -------------------------------------------------------------------------------- /tests/in/training/minimum-1.inp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/tests/in/training/minimum-1.inp -------------------------------------------------------------------------------- /tests/in/training/minimum-2.inp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/tests/in/training/minimum-2.inp -------------------------------------------------------------------------------- /tests/in/training/target-channel-memory.inp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/tests/in/training/target-channel-memory.inp -------------------------------------------------------------------------------- /tests/in/training/traj-gru.inp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/tests/in/training/traj-gru.inp -------------------------------------------------------------------------------- /tests/ref/attention-channel-memory.inp.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/tests/ref/attention-channel-memory.inp.pth -------------------------------------------------------------------------------- /tests/ref/attention.inp.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/tests/ref/attention.inp.pth -------------------------------------------------------------------------------- /tests/ref/channel-infusion.inp.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/tests/ref/channel-infusion.inp.pth -------------------------------------------------------------------------------- /tests/ref/channel-memory.inp.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/tests/ref/channel-memory.inp.pth -------------------------------------------------------------------------------- /tests/ref/conv-lstm.inp.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/tests/ref/conv-lstm.inp.pth -------------------------------------------------------------------------------- /tests/ref/json-input.inp.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/tests/ref/json-input.inp.pth -------------------------------------------------------------------------------- /tests/ref/minimum-1.inp.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/tests/ref/minimum-1.inp.pth -------------------------------------------------------------------------------- /tests/ref/minimum-1_infilled.nc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/tests/ref/minimum-1_infilled.nc -------------------------------------------------------------------------------- /tests/ref/minimum-2.inp.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/tests/ref/minimum-2.inp.pth -------------------------------------------------------------------------------- /tests/ref/target-channel-memory.inp.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/tests/ref/target-channel-memory.inp.pth -------------------------------------------------------------------------------- /tests/ref/traj-gru.inp.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/tests/ref/traj-gru.inp.pth -------------------------------------------------------------------------------- /tests/test_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/tests/test_evaluation.py -------------------------------------------------------------------------------- /tests/test_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FREVA-CLINT/climatereconstructionAI/HEAD/tests/test_training.py --------------------------------------------------------------------------------