engine-maintenance-classifier / model_metadata.json
imanandshah's picture
Upload model_metadata.json with huggingface_hub
e9f6655 verified
Raw History Blame Contribute Delete
797 Bytes
{
"artifact": "engine_maintenance_pipeline.joblib",
"pipeline": [
"IQRCapper",
"StandardScaler",
"GradientBoostingClassifier"
],
"requires_module": "preprocessing.py",
"input_features": [
"Engine_RPM",
"Lub_Oil_Pressure",
"Fuel_Pressure",
"Coolant_Pressure",
"Lub_Oil_Temperature",
"Coolant_Temperature"
],
"positive_class": 1,
"class_names": {
"0": "Normal",
"1": "Requires Maintenance"
},
"best_params": {
"learning_rate": 0.05,
"max_depth": 2,
"n_estimators": 120,
"subsample": 1.0
},
"metrics": {
"Train_Accuracy": 0.6774,
"Test_Accuracy": 0.6627,
"Balanced_Accuracy": 0.5922,
"Precision": 0.6845,
"Recall": 0.8624,
"F1": 0.7632,
"ROC_AUC": 0.7017
},
"model_version": "1.0.0"
}