GRIB 2 output

neural_lam/data_config.yaml

@@ -1,64 +1,70 @@
 dataset:
-  name: meps_example
+  name: cosmo
+
+  # Value definitions
   var_names:
-    - pres_0g
-    - pres_0s
-    - nlwrs_0
-    - nswrs_0
-    - r_2
-    - r_65
-    - t_2
-    - t_65
-    - t_500
-    - t_850
-    - u_65
-    - u_850
-    - v_65
-    - v_850
-    - wvint_0
-    - z_1000
-    - z_500
+    - "T"
+    - "U"
+    - "V"
+    - "RELHUM"
+    - "PMSL"
+    - "PP"
   var_units:
-    - Pa
-    - Pa
-    - r"$\mathrm{W}/\mathrm{m}^2$"
-    - r"$\mathrm{W}/\mathrm{m}^2$"
-    - ""
-    - ""
-    - K
-    - K
     - K
-    - K
-    - m/s
     - m/s
     - m/s
-    - m/s
-    - r"$\mathrm{kg}/\mathrm{m}^2$"
-    - r"$\mathrm{m}^2/\mathrm{s}^2$"
-    - r"$\mathrm{m}^2/\mathrm{s}^2$"
+    - Perc.
+    - Pa
+    - hPa
   var_longnames:
-    - pres_heightAboveGround_0_instant
-    - pres_heightAboveSea_0_instant
-    - nlwrs_heightAboveGround_0_accum
-    - nswrs_heightAboveGround_0_accum
-    - r_heightAboveGround_2_instant
-    - r_hybrid_65_instant
-    - t_heightAboveGround_2_instant
-    - t_hybrid_65_instant
-    - t_isobaricInhPa_500_instant
-    - t_isobaricInhPa_850_instant
-    - u_hybrid_65_instant
-    - u_isobaricInhPa_850_instant
-    - v_hybrid_65_instant
-    - v_isobaricInhPa_850_instant
-    - wvint_entireAtmosphere_0_instant
-    - z_isobaricInhPa_1000_instant
-    - z_isobaricInhPa_500_instant
+    - "Temperature"
+    - "Zonal wind component"
+    - "Meridional wind component"
+    - "Relative humidity"
+    - "Pressure at Mean Sea Level"
+    - "Pressure Perturbation"
+  var_is_3d:
+    - 1
+    - 1
+    - 1
+    - 1
+    - 0
+    - 1
+  grib_names:
+    PP: "pres"
+    QV: "q"
+    RELHUM: "r"
+    T: "t"
+    U: "u"
+    V: "v"
+    W: "wz"
+    CLCT: "ccl"
+    PMSL: "prmsl"
+    PS: "sp"
+    T_2M: "2t"
+    TOT_PREC: "tp"
+    U_10M: "10u"
+    V_10M: "10v"
+  vertical_levels: [1, 5, 13, 22, 38, 41, 60]
   num_forcing_features: 16
-grid_shape_state: [268, 238]
-projection:
-  class: LambertConformal
-  kwargs:
-    central_longitude: 15.0
-    central_latitude: 63.3
-    standard_parallels: [63.3, 63.3]
+
+  # Plotting 
+  eval_plot_vars: ["TQV"]
+  grid_shape_state: [390, 582]
+  projection:
+    class: LambertConformal
+    kwargs:
+      central_longitude: 15.0
+      central_latitude: 63.3
+      standard_parallels: [63.3, 63.3]
+  sample_grib:
+    "templates/lfff02180000"
+  sample_z_grib:
+   "templates/lfff02180000z"
+  eval_datetime:
+    ["2020100215"]
+
+  # Time step prediction during training / prediction (eval)
+  train_horizon: 3
+  eval_horizon: 25
+