File size: 2,250 Bytes
ca783e4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 | {
"architectures": [
"WeatherForcastModel"
],
"auto_map": {
"AutoConfig": "configuration.WeatherModelConfig",
"AutoModel": "modeling.WeatherForcastModel"
},
"continuous_loss_weight": 1.0,
"dropout": 0.1,
"dtype": "float32",
"hidden_dim": 384,
"input_dim": 22,
"location_emb_dim": 32,
"model_type": "mwm",
"num_layers": 6,
"num_locations": 82,
"num_predict": 12,
"num_weather_classes": 7,
"rain_loss_weight": 0.5,
"rain_pos_weight": 6.548002378792392,
"seq_len": 72,
"target_norms": {
"apparent": {
"mean": 16.41952039071746,
"std": 12.332994012430817,
"transform": "raw"
},
"cloud_cover": {
"mean": 51.713692581765955,
"std": 42.028681460392654,
"transform": "raw"
},
"humidity": {
"mean": 69.18757318577516,
"std": 21.523738585258304,
"transform": "raw"
},
"precip": {
"mean": 0.056842080769016465,
"std": 0.22550550790216348,
"transform": "log1p"
},
"sea_level_pressure": {
"mean": 1014.5678436112748,
"std": 7.45947258595889,
"transform": "raw"
},
"surface_pressure": {
"mean": 963.4382703481587,
"std": 86.3934876131444,
"transform": "raw"
},
"temp": {
"mean": 16.863310769348754,
"std": 9.930158388118258,
"transform": "raw"
},
"wind": {
"mean": 9.9528978678247,
"std": 6.676634456702914,
"transform": "raw"
},
"wind_dir_cos": {
"mean": -0.033494943023861114,
"std": 0.7051367654528827,
"transform": "raw"
},
"wind_dir_sin": {
"mean": -0.0021146874377944323,
"std": 0.7082766124425719,
"transform": "raw"
}
},
"transformers_version": "5.5.0",
"use_cache": false,
"weather_class_weights": [
0.10000000149011612,
0.10000000149011612,
1.9490022659301758,
0.12914901971817017,
0.48891985416412354,
1.7922818660736084,
2.5796220302581787
],
"weather_log_priors": [
-1.050954818725586,
-0.6768107414245605,
-5.038214683532715,
-2.3241090774536133,
-3.6553406715393066,
-4.9543867111206055,
-5.318540096282959
],
"weather_loss_weight": 1.5,
"weather_prior_tau": 0.0
}
|