predictive-maintenance-model / model_metadata.json
nithyanatarajan's picture
Upload model_metadata.json with huggingface_hub
c0e3256 verified
raw
history blame contribute delete
706 Bytes
{
"model_name": "Random Forest",
"features": [
"engine_rpm",
"lub_oil_pressure",
"fuel_pressure",
"coolant_pressure",
"lub_oil_temp",
"coolant_temp",
"rpm_x_fuel_pressure",
"rpm_bins",
"oil_health_index"
],
"target": "engine_condition",
"threshold": 0.4,
"test_metrics": {
"recall": 0.9420995670995671,
"f2": 0.8666002986560478,
"f1": 0.7736058653632526,
"precision": 0.6562382208820203,
"accuracy": 0.6523370863186626
},
"best_params": {
"classifier__max_depth": null,
"classifier__max_features": "sqrt",
"classifier__min_samples_leaf": 1,
"classifier__min_samples_split": 2,
"classifier__n_estimators": 500
}
}