akshatshaw's picture
Add model config
1b039c6 verified
{
"model_type": "FrontLSTM",
"input_size_sta": 19,
"input_size_dyn": 40,
"hidden_size": 64,
"output_size": 1,
"num_layers": 2,
"drop_prob": 0.3,
"static_features": [
"prcp(mm/day)",
"tmax(C)",
"tmin(C)",
"tavg(C)",
"srad_lw(w/m2)",
"srad_sw(w/m2)",
"wind_u(m/s)",
"wind_v(m/s)",
"wind(m/s)",
"rel_hum(%)",
"pet(mm/day)",
"pet_gleam(mm/day)",
"aet_gleam(mm/day)",
"evap_canopy(mm/day)",
"evap_surface(mm/day)",
"sm_lvl1(kg/m2)",
"sm_lvl2(kg/m2)",
"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": 473.00732421875,
"test_mae": 143.65219116210938,
"test_r2": 0.4109150744583362
}