File size: 732 Bytes
1efc8e5 | 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 | {
"best_model": "RandomForest",
"best_accuracy": 0.8021,
"best_f1_weighted": 0.8021,
"rf_accuracy": 0.8021,
"rf_f1_weighted": 0.8021,
"xgb_accuracy": 0.7323,
"xgb_f1_weighted": 0.7365,
"features": [
"pm2_5_ugm3",
"pm10_ugm3",
"co_ugm3",
"no2_ugm3",
"so2_ugm3",
"o3_ugm3",
"month",
"hour"
],
"target_int": "aqi_category_enc",
"target_str": "aqi_category",
"categories": [
"Good",
"Satisfactory",
"Moderate",
"Poor",
"Very Poor",
"Severe"
],
"int_to_category": {
"0": "Good",
"1": "Satisfactory",
"2": "Moderate",
"3": "Poor",
"4": "Very Poor",
"5": "Severe"
},
"classes_present": [
0,
2,
3,
4,
5
]
} |