feat(constants): add GFS statistics to NWPStatDict (#145)

openclimatefix · Jan 24, 2025 · bc43997 · bc43997
1 parent b6c5321
commit bc43997
Showing 1 changed file with 48 additions and 0 deletions.
diff --git a/ocf_data_sampler/constants.py b/ocf_data_sampler/constants.py
@@ -5,6 +5,7 @@
 NWP_PROVIDERS = [
     "ukv",
     "ecmwf",
+    "gfs"
 ]
 
 
@@ -127,13 +128,60 @@ def __getitem__(self, key):
 ECMWF_STD = _to_data_array(ECMWF_STD)
 ECMWF_MEAN = _to_data_array(ECMWF_MEAN)
 
+# ------ GFS
+GFS_STD = {
+    "dlwrf": 96.305916,
+    "dswrf": 246.18533,
+    "hcc": 42.525383,
+    "lcc": 44.3732,
+    "mcc": 43.150745,
+    "prate": 0.00010159573,
+    "r": 25.440672,
+    "sde": 0.43345627,
+    "t": 22.825893,
+    "tcc": 41.030598,
+    "u10": 5.470838,
+    "u100": 6.8899174,
+    "v10": 4.7401133,
+    "v100": 6.076132,
+    "vis": 8294.022,
+    "u": 10.614556,
+    "v": 7.176398,
+}
+
+GFS_MEAN = {
+    "dlwrf": 298.342,
+    "dswrf": 168.12321,
+    "hcc": 35.272,
+    "lcc": 43.578342,
+    "mcc": 33.738823,
+    "prate": 2.8190969e-05,
+    "r": 18.359747,
+    "sde": 0.36937004,
+    "t": 278.5223,
+    "tcc": 66.841606,
+    "u10": -0.0022310058,
+    "u100": 0.0823025,
+    "v10": 0.06219831,
+    "v100": 0.0797807,
+    "vis": 19628.32,
+    "u": 11.645444,
+    "v": 0.12330122,
+}
+
+GFS_STD = _to_data_array(GFS_STD)
+GFS_MEAN = _to_data_array(GFS_MEAN)
+
+
 NWP_STDS = NWPStatDict(
     ukv=UKV_STD,
     ecmwf=ECMWF_STD,
+    gfs=GFS_STD
 )
 NWP_MEANS = NWPStatDict(
     ukv=UKV_MEAN,
     ecmwf=ECMWF_MEAN,
+    gfs=GFS_MEAN
 )
 
 # ------ Satellite