├── API ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-38.pyc │ ├── dataloader.cpython-38.pyc │ ├── dataloader_moving_mnist.cpython-38.pyc │ ├── dataloader_ns2d.cpython-38.pyc │ ├── dataloader_sevir.cpython-38.pyc │ ├── dataloader_taxibj.cpython-38.pyc │ ├── dataloader_taxibj_12_12.cpython-38.pyc │ ├── dataloader_weather.cpython-38.pyc │ ├── metrics.cpython-38.pyc │ └── recorder.cpython-38.pyc ├── dataloader.py ├── dataloader_sevir.py ├── dataloader_taxibj.py ├── dataloader_taxibj_12_12.py ├── dataloader_weather.py ├── metrics.py └── recorder.py ├── README.md ├── __pycache__ ├── dataloader.cpython-38.pyc └── nvwa_pretrain.cpython-38.pyc ├── attention_map ├── Attention_map.npy └── casual_data_aug.ipynb ├── causal_test.ipynb ├── checkpoints └── 200.pth ├── dataloader.py ├── main.py ├── mask.py ├── metrics.py ├── models ├── __init__.py ├── convlstm_model.py ├── e3dlstm_model.py ├── mau_model.py ├── mim_model.py ├── mmvp_model.py ├── phydnet_model.py ├── predrnn_model.py ├── predrnnpp_model.py ├── predrnnv2_model.py ├── simvp_model.py └── swinlstm_model.py ├── nvwa_downstream_pred.py ├── nvwa_upstream_pretrain.py ├── pretrain_model_save ├── 2024big_best_model.pth └── big_best_model.pth ├── results ├── labels.npy └── preds.npy ├── train_prediction.py ├── train_reconstruction.py ├── utils.py └── vit.ipynb /API/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/API/__init__.py -------------------------------------------------------------------------------- /API/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/API/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /API/__pycache__/dataloader.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/API/__pycache__/dataloader.cpython-38.pyc -------------------------------------------------------------------------------- /API/__pycache__/dataloader_moving_mnist.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/API/__pycache__/dataloader_moving_mnist.cpython-38.pyc -------------------------------------------------------------------------------- /API/__pycache__/dataloader_ns2d.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/API/__pycache__/dataloader_ns2d.cpython-38.pyc -------------------------------------------------------------------------------- /API/__pycache__/dataloader_sevir.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/API/__pycache__/dataloader_sevir.cpython-38.pyc -------------------------------------------------------------------------------- /API/__pycache__/dataloader_taxibj.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/API/__pycache__/dataloader_taxibj.cpython-38.pyc -------------------------------------------------------------------------------- /API/__pycache__/dataloader_taxibj_12_12.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/API/__pycache__/dataloader_taxibj_12_12.cpython-38.pyc -------------------------------------------------------------------------------- /API/__pycache__/dataloader_weather.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/API/__pycache__/dataloader_weather.cpython-38.pyc -------------------------------------------------------------------------------- /API/__pycache__/metrics.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/API/__pycache__/metrics.cpython-38.pyc -------------------------------------------------------------------------------- /API/__pycache__/recorder.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/API/__pycache__/recorder.cpython-38.pyc -------------------------------------------------------------------------------- /API/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/API/dataloader.py -------------------------------------------------------------------------------- /API/dataloader_sevir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/API/dataloader_sevir.py -------------------------------------------------------------------------------- /API/dataloader_taxibj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/API/dataloader_taxibj.py -------------------------------------------------------------------------------- /API/dataloader_taxibj_12_12.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/API/dataloader_taxibj_12_12.py -------------------------------------------------------------------------------- /API/dataloader_weather.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/API/dataloader_weather.py -------------------------------------------------------------------------------- /API/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/API/metrics.py -------------------------------------------------------------------------------- /API/recorder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/API/recorder.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/README.md -------------------------------------------------------------------------------- /__pycache__/dataloader.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/__pycache__/dataloader.cpython-38.pyc -------------------------------------------------------------------------------- /__pycache__/nvwa_pretrain.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/__pycache__/nvwa_pretrain.cpython-38.pyc -------------------------------------------------------------------------------- /attention_map/Attention_map.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/attention_map/Attention_map.npy -------------------------------------------------------------------------------- /attention_map/casual_data_aug.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/attention_map/casual_data_aug.ipynb -------------------------------------------------------------------------------- /causal_test.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/causal_test.ipynb -------------------------------------------------------------------------------- /checkpoints/200.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/checkpoints/200.pth -------------------------------------------------------------------------------- /dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/dataloader.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/main.py -------------------------------------------------------------------------------- /mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/mask.py -------------------------------------------------------------------------------- /metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/metrics.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/convlstm_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/models/convlstm_model.py -------------------------------------------------------------------------------- /models/e3dlstm_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/models/e3dlstm_model.py -------------------------------------------------------------------------------- /models/mau_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/models/mau_model.py -------------------------------------------------------------------------------- /models/mim_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/models/mim_model.py -------------------------------------------------------------------------------- /models/mmvp_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/models/mmvp_model.py -------------------------------------------------------------------------------- /models/phydnet_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/models/phydnet_model.py -------------------------------------------------------------------------------- /models/predrnn_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/models/predrnn_model.py -------------------------------------------------------------------------------- /models/predrnnpp_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/models/predrnnpp_model.py -------------------------------------------------------------------------------- /models/predrnnv2_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/models/predrnnv2_model.py -------------------------------------------------------------------------------- /models/simvp_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/models/simvp_model.py -------------------------------------------------------------------------------- /models/swinlstm_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/models/swinlstm_model.py -------------------------------------------------------------------------------- /nvwa_downstream_pred.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/nvwa_downstream_pred.py -------------------------------------------------------------------------------- /nvwa_upstream_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/nvwa_upstream_pretrain.py -------------------------------------------------------------------------------- /pretrain_model_save/2024big_best_model.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/pretrain_model_save/2024big_best_model.pth -------------------------------------------------------------------------------- /pretrain_model_save/big_best_model.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/pretrain_model_save/big_best_model.pth -------------------------------------------------------------------------------- /results/labels.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/results/labels.npy -------------------------------------------------------------------------------- /results/preds.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/results/preds.npy -------------------------------------------------------------------------------- /train_prediction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/train_prediction.py -------------------------------------------------------------------------------- /train_reconstruction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/train_reconstruction.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/utils.py -------------------------------------------------------------------------------- /vit.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alexander-wu/NuwaDynamics/HEAD/vit.ipynb --------------------------------------------------------------------------------