The source code of the SOTA Dynamic Graph Reformer (DGFormer) model for weather station prediction.

DGFormer is the update of Dynamic Climate Pattern Graph Recurrent Network (DCPGRN) for meteorological forecasting This is a official pyTorch implementation of DGFormer

torch>=1.7.0

torch-geometric-temporal

Dependency can be installed using the following command:

pip install -r requirements.txt

You can download our data from google drive , and put it into ./data dir.

Download the dataset and copy it into data/ dir. And Unzip them.

To execute the baseline experiment, you can change the "graph_model",

The configuration is set in config.yaml file for training process. Run the following commands to train the target model.

python train.py

We are further developing the distributed version for a larger scale GNN model.

We compared our model with STGCN, DCRNN, AGCRN, WGC, Dysat, TGAT, GWN and CLCRN, also with ECMWF (NWP method). The reslut shows that our model outperform others especially in temperature prediction

The project is developed based on PM2.5-GNN, Reformer and IDGL for dynamic graph.