Add model config

config.json

@@ -1,12 +1,63 @@
 {
   "model_type": "FrontLSTM",
-  "input_size_sta": 19,
-  "input_size_dyn": 40,
+  "input_size_sta": 49,
+  "input_size_dyn": 18,
   "hidden_size": 512,
   "output_size": 1,
   "num_layers": 2,
-  "drop_prob": 0.3,
+  "drop_prob": 0.1,
   "static_features": [
+    "elev_mean",
+    "slope_mean",
+    "ghi_area",
+    "dpsbar",
+    "freq_q_low",
+    "freq_q_high",
+    "p_mean",
+    "mean_anum_flow",
+    "bfi",
+    "runoff_ratio",
+    "p_mean_anum",
+    "q_mean",
+    "mean_swmn_flow",
+    "mean_atmn_flow",
+    "mean_wint_flow",
+    "mean_sumr_flow",
+    "high_prec_freq",
+    "low_prec_freq",
+    "max_high_prec_dur",
+    "max_low_prec_dur",
+    "aridity_p_pet",
+    "water_frac",
+    "trees_frac",
+    "flooded_veg_frac",
+    "crops_frac",
+    "built_area_frac",
+    "lai_mean",
+    "soil_depth",
+    "soil_awc_top",
+    "soil_awc_sub",
+    "soil_awsc_min",
+    "soil_awsc_max",
+    "sand_frac_top",
+    "sand_frac_sub",
+    "silt_frac_top",
+    "silt_frac_sub",
+    "clay_frac_top",
+    "clay_frac_sub",
+    "gravel_frac_top",
+    "gravel_frac_sub",
+    "bulkdense_top_mean",
+    "bulkdens_sub_mean",
+    "wtd",
+    "org_carb_sub_mean",
+    "org_carb_top_mean",
+    "geol_porosity",
+    "geol_permeability",
+    "total_storage",
+    "reservoir_index"
+  ],
+  "dynamic_features": [
     "prcp(mm/day)",
     "tmax(C)",
     "tmin(C)",
@@ -17,7 +68,6 @@
     "wind_v(m/s)",
     "wind(m/s)",
     "rel_hum(%)",
-    "pet(mm/day)",
     "pet_gleam(mm/day)",
     "aet_gleam(mm/day)",
     "evap_canopy(mm/day)",
@@ -27,49 +77,7 @@
     "sm_lvl3(kg/m2)",
     "sm_lvl4(kg/m2)"
   ],
-  "dynamic_features": [
-    "umap_40d_dim_1",
-    "umap_40d_dim_2",
-    "umap_40d_dim_3",
-    "umap_40d_dim_4",
-    "umap_40d_dim_5",
-    "umap_40d_dim_6",
-    "umap_40d_dim_7",
-    "umap_40d_dim_8",
-    "umap_40d_dim_9",
-    "umap_40d_dim_10",
-    "umap_40d_dim_11",
-    "umap_40d_dim_12",
-    "umap_40d_dim_13",
-    "umap_40d_dim_14",
-    "umap_40d_dim_15",
-    "umap_40d_dim_16",
-    "umap_40d_dim_17",
-    "umap_40d_dim_18",
-    "umap_40d_dim_19",
-    "umap_40d_dim_20",
-    "umap_40d_dim_21",
-    "umap_40d_dim_22",
-    "umap_40d_dim_23",
-    "umap_40d_dim_24",
-    "umap_40d_dim_25",
-    "umap_40d_dim_26",
-    "umap_40d_dim_27",
-    "umap_40d_dim_28",
-    "umap_40d_dim_29",
-    "umap_40d_dim_30",
-    "umap_40d_dim_31",
-    "umap_40d_dim_32",
-    "umap_40d_dim_33",
-    "umap_40d_dim_34",
-    "umap_40d_dim_35",
-    "umap_40d_dim_36",
-    "umap_40d_dim_37",
-    "umap_40d_dim_38",
-    "umap_40d_dim_39",
-    "umap_40d_dim_40"
-  ],
-  "test_rmse": 601.3206787109375,
-  "test_mae": 169.7910614013672,
-  "test_r2": 0.048280939963296965
+  "test_rmse": 361.0433044433594,
+  "test_mae": 100.75887298583984,
+  "test_r2": 0.6568179124833937
 }