{
  "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
}