predictive-maintenance-model / model_metadata.json
jskswamy's picture
Upload model_metadata.json with huggingface_hub
3553773 verified
{
"model_name": "AdaBoost (Optuna)",
"algorithm": "AdaBoost",
"tuning_method": "Optuna",
"features": [
"Engine RPM",
"Lub Oil Pressure",
"Fuel Pressure",
"Coolant Pressure",
"Lub Oil Temp",
"Coolant Temp",
"RPM_x_OilPressure",
"RPM_x_FuelPressure",
"RPM_x_CoolantPressure",
"OilTemp_x_OilPressure",
"CoolantTemp_x_CoolantPressure",
"RPM_squared",
"OilPressure_squared",
"TempDiff",
"OilFuelPressureRatio",
"CoolantOilPressureRatio",
"OilHealthIndex",
"CoolantStress",
"OilTempPerRPM",
"CoolantTempPerRPM",
"PressureSum",
"TempSum"
],
"n_features": 22,
"target": "Engine Condition",
"threshold": 0.3699999999999999,
"test_metrics": {
"balanced_accuracy": 0.5159377210970009,
"roc_auc": 0.6980806781707058,
"recall": 0.9875541125541125,
"precision": 0.6381118881118881,
"f2": 0.8900702301989856
},
"best_params": {
"base_max_depth": "3",
"n_estimators": "383",
"learning_rate": "0.2608158369431769"
},
"training_info": {
"train_samples": 14650,
"val_samples": 1954,
"test_samples": 2931
}
}