nielsr HF Staff commited on
Commit
f732c5e
·
verified ·
1 Parent(s): 912fe5a

Add pipeline tag, library name, paper link and Github repo link

Browse files

This PR improves the model card by:
- Setting the pipeline tag to `time-series-forecasting`
- Setting the library_name to `torch` since the model uses PyTorch code
- Adding links to the paper and github

Files changed (1) hide show
  1. README.md +77 -3
README.md CHANGED
@@ -1,3 +1,77 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ pipeline_tag: time-series-forecasting
4
+ library_name: torch
5
+ ---
6
+
7
+ # OneForecast: A Universal Framework for Global and Regional Weather Forecasting
8
+
9
+ This repository contains the code for the model described in the paper [OneForecast: A Universal Framework for Global and Regional Weather Forecasting](https://huggingface.co/papers/2502.00338).
10
+
11
+ Github repository: https://github.com/YuanGao-YG/OneForecast.
12
+
13
+ Accurate weather forecasts are important for disaster prevention, agricultural planning, etc. Traditional numerical weather prediction (NWP) methods offer physically interpretable high-accuracy predictions but are computationally expensive and fail to fully leverage rapidly growing historical data. In recent years, deep learning models have made significant progress in weather forecasting, but challenges remain, such as balancing global and regional high-resolution forecasts, excessive smoothing in extreme event predictions, and insufficient dynamic system modeling. To address these issues, this paper proposes a global-regional nested weather forecasting framework (OneForecast) based on graph neural networks. By combining a dynamic system perspective with multi-grid theory, we construct a multi-scale graph structure and densify the target region to capture local high-frequency features. We introduce an adaptive messaging mechanism, using dynamic gating units to deeply integrate node and edge features for more accurate extreme event forecasting. For high-resolution regional forecasts, we propose a neural nested grid method to mitigate boundary information loss. Experimental results show that OneForecast performs excellently across global to regional scales and short-term to long-term forecasts, especially in extreme event predictions.
14
+
15
+ # Quick Start
16
+
17
+ ### Installation
18
+
19
+ - cuda 11.8
20
+
21
+ ```
22
+ # git clone this repository
23
+ git clone https://github.com/YuanGao-YG/OneForecast.git
24
+ cd OneForecast
25
+
26
+ # create new anaconda env
27
+ conda env create -f environment.yml
28
+ conda activate oneforecast
29
+ ```
30
+
31
+
32
+ ### Inference
33
+
34
+ 1. Global Forecasts Inference
35
+
36
+ (1) Preparing the test data as follows:
37
+
38
+ ```
39
+ ./data/
40
+ |--global
41
+ | |--test
42
+ | | |--2020.h5
43
+ | |--mean.npy
44
+ | |--std.npy
45
+ ```
46
+
47
+ (2) Inference with 1-step supervised pretrained ckpt:
48
+ ```
49
+ sh inference.sh
50
+ ```
51
+
52
+ (3) Inference with finetuned pretrained ckpt:
53
+ ```
54
+ sh inference_finetune.sh
55
+ ```
56
+
57
+ 2. Regional Forecasts Inference
58
+
59
+ (1) Preparing the test data as follows:
60
+
61
+ ```
62
+ ./data/
63
+ |--global
64
+ | |--test
65
+ | | |--2020.h5
66
+ | |--mean.npy
67
+ | |--std.npy
68
+ |--regional
69
+ | |--test
70
+ | | |--2020.h5
71
+ ```
72
+
73
+ (2) Inference with 1-step supervised pretrained ckpt:
74
+
75
+ ```
76
+ sh inference_nng.sh
77
+ ```