vayu-models / classifier /classifier_metadata.json
rachitgoyell's picture
Upload 12 files
1efc8e5 verified
raw
history blame contribute delete
732 Bytes
{
"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
]
}