SupraWeather1.5-Small / config.json
AxionLab-official's picture
Upload 4 files
31462fd verified
Raw
History Blame Contribute Delete
3.2 kB
{
"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"
}