Modelpublic

indianakhil/engine-predictive-maintenance-model

sourceHugging Facemitupdated 6mo agoView on Hugging Face

1likes

Model Card

Engine Predictive Maintenance Model

Model Description

Binary classifier predicting engine health (Normal vs Faulty) from six sensor readings.

—Model Type: AdaBoostClassifier
—Task: Binary Classification (0=Normal, 1=Faulty)
—Training Data: indianakhil/engine-predictive-maintenance (19,535 records)
—Best Hyperparameters: {'learning_rate': 0.5, 'n_estimators': 50}

Performance (Test Set — 20% holdout)

Metric	Score
Accuracy	0.6644
Precision	0.6787
Recall	0.8883
F1-Score	0.7695
ROC-AUC	0.6960
CV F1 (5-fold)	0.7663

Input Features

EngineRPM, LubOilPressure, FuelPressure, CoolantPressure, LubOilTemperature, CoolantTemperature

Usage

python

from huggingface_hub import hf_hub_download
import joblib, pandas as pd
model = joblib.load(hf_hub_download(
    repo_id='indianakhil/engine-predictive-maintenance-model',
    filename='best_model.pkl'))
pred = model.predict(X)  # 0=Normal, 1=Faulty
prob = model.predict_proba(X)[:, 1]  # Fault probability