{
  "architectures": [
    "SupraWeatherModel"
  ],
  "attn_dropout": 0.15,
  "cat_cardinalities": [
    8,
    12,
    6,
    6,
    5
  ],
  "categorical_features": [
    "wind_direction",
    "month",
    "air_mass",
    "climate_zone",
    "terrain_type"
  ],
  "continuous_features": [
    "temperature",
    "humidity",
    "pressure",
    "pressure_trend",
    "wind_speed",
    "altitude",
    "cloud_cover",
    "visibility",
    "solar_radiation",
    "dew_point",
    "heat_index",
    "wind_chill",
    "storm_index",
    "rain_potential",
    "instability_index"
  ],
  "d_token": 256,
  "dtype": "float32",
  "ffn_dropout": 0.1,
  "ffn_factor": 1.333,
  "id2label": {
    "0": "clear",
    "1": "partly_cloudy",
    "2": "cloudy",
    "3": "overcast",
    "4": "mist",
    "5": "fog",
    "6": "dense_fog",
    "7": "light_rain",
    "8": "rain",
    "9": "heavy_rain",
    "10": "torrential_rain",
    "11": "thunderstorm",
    "12": "severe_thunderstorm",
    "13": "snow",
    "14": "heavy_snow",
    "15": "blizzard",
    "16": "freezing_rain",
    "17": "ice_storm",
    "18": "soft_hail",
    "19": "sleet",
    "20": "cold_front",
    "21": "heat_wave",
    "22": "cold_wave",
    "23": "windstorm",
    "24": "dust_storm"
  },
  "label2id": {
    "blizzard": 15,
    "clear": 0,
    "cloudy": 2,
    "cold_front": 20,
    "cold_wave": 22,
    "dense_fog": 6,
    "dust_storm": 24,
    "fog": 5,
    "freezing_rain": 16,
    "heat_wave": 21,
    "heavy_rain": 9,
    "heavy_snow": 14,
    "ice_storm": 17,
    "light_rain": 7,
    "mist": 4,
    "overcast": 3,
    "partly_cloudy": 1,
    "rain": 8,
    "severe_thunderstorm": 12,
    "sleet": 19,
    "snow": 13,
    "soft_hail": 18,
    "thunderstorm": 11,
    "torrential_rain": 10,
    "windstorm": 23
  },
  "means": {
    "altitude": 561.1331787109375,
    "cloud_cover": 75.88937377929688,
    "dew_point": 10.450345993041992,
    "heat_index": 24.373546600341797,
    "humidity": 73.68315887451172,
    "instability_index": 9.872715950012207,
    "pressure": 953.83740234375,
    "pressure_trend": -0.0021683243103325367,
    "rain_potential": 68.46627807617188,
    "solar_radiation": 342.4497375488281,
    "storm_index": 7.1922607421875,
    "temperature": 15.711910247802734,
    "visibility": 9.542573928833008,
    "wind_chill": 14.176212310791016,
    "wind_speed": 12.290905952453613
  },
  "model_type": "supra_weather_ft",
  "n_blocks": 4,
  "n_heads": 8,
  "num_continuous": 15,
  "residual_dropout": 0.0,
  "stds": {
    "altitude": 524.8400268554688,
    "cloud_cover": 25.562358856201172,
    "dew_point": 18.99576187133789,
    "heat_index": 36.05759048461914,
    "humidity": 14.650811195373535,
    "instability_index": 7.713136196136475,
    "pressure": 48.49877166748047,
    "pressure_trend": 3.3718252182006836,
    "rain_potential": 34.90696716308594,
    "solar_radiation": 179.88714599609375,
    "storm_index": 3.5440399646759033,
    "temperature": 17.988454818725586,
    "visibility": 7.803797721862793,
    "wind_chill": 20.287734985351562,
    "wind_speed": 7.582037448883057
  },
  "transformers_version": "5.0.0",
  "use_cache": false,
  "_name_or_path": "SupraLabs/SupraWeather-Nano-1.5"
}