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