akshatshaw's picture
Add model config
522a719 verified
{
"model_type": "FrontLSTM",
"input_size_sta": 49,
"input_size_dyn": 18,
"hidden_size": 512,
"output_size": 1,
"num_layers": 2,
"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)",
"tavg(C)",
"srad_lw(w/m2)",
"srad_sw(w/m2)",
"wind_u(m/s)",
"wind_v(m/s)",
"wind(m/s)",
"rel_hum(%)",
"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)"
],
"test_rmse": 361.0433044433594,
"test_mae": 100.75887298583984,
"test_r2": 0.6568179124833937
}