Hugging Face's logo

SharleyK
/
predictive-maintenance-model

Scikit-learn
predictive-maintenance
engine-failure-prediction
adaboost
classification
Model card Files
xet
Community
predictive-maintenance-model
File size: 2,741 Bytes 
3bed759
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
8032aa1
3bed759
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
 
---
license: mit
tags:
- predictive-maintenance
- engine-failure-prediction
- adaboost
- classification
library_name: sklearn
---

# Predictive Maintenance Model - Engine Failure Prediction

## Model Description

This model predicts engine failures for automotive predictive maintenance using sensor data.

**Model Type:** AdaBoost
**Task:** Binary Classification (Normal vs Faulty Engine)
**Framework:** scikit-learn / XGBoost

## Model Performance

### Test Set Metrics
- **Accuracy:** 0.6668
- **Precision:** 0.6854
- **Recall:** 0.8713 (Primary metric - minimizes false negatives)
- **F1-Score:** 0.7673
- **ROC-AUC:** 0.6959

## Model Details

### Hyperparameters
```python
{
  "learning_rate": 0.05,
  "n_estimators": 100
}
```

### Training Information
- **Training Samples:** 15,628
- **Test Samples:** 3,907
- **Features:** 17
- **Training Date:** 2026-02-04 15:04:16

## Features

The model uses 17 features including:
- Engine RPM
- Lubricating oil pressure and temperature
- Fuel pressure
- Coolant pressure and temperature
- Engineered features (temperature-pressure ratios, differentials, etc.)

## Usage

```python
import joblib
from huggingface_hub import hf_hub_download

# Download model
model_path = hf_hub_download(
    repo_id="SharleyK/predictive-maintenance-model",
    filename="best_model.pkl"
)

# Load model
model = joblib.load(model_path)

# Download scaler
scaler_path = hf_hub_download(
    repo_id="SharleyK/predictive-maintenance-model",
    filename="scaler.pkl"
)
scaler = joblib.load(scaler_path)

# Make predictions
X_new_scaled = scaler.transform(X_new)
predictions = model.predict(X_new_scaled)
probabilities = model.predict_proba(X_new_scaled)

# Interpret results
# 0 = Normal/Healthy Engine
# 1 = Faulty/Requires Maintenance
```

## Model Selection

This model was selected from 6 candidates:
- Decision Tree
- Bagging Classifier
- Random Forest
- AdaBoost
- Gradient Boosting
- XGBoost

Selection criteria: Highest test recall (to minimize false negatives - missed failures)

## Business Impact

- Reduces unplanned breakdowns by detecting failures early
- Minimizes emergency repair costs
- Optimizes maintenance scheduling
- Improves fleet availability and safety

## Limitations

- Requires all sensor inputs to be available
- Trained on specific engine types (automotive and small engines)
- Performance may degrade if sensor calibration changes
- Requires periodic retraining with new data

## Citation

```
@model{predictive_maintenance_engine_model,
  author = {SharleyK},
  title = {Predictive Maintenance Model - Engine Failure Prediction},
  year = {2026},
  publisher = {Hugging Face},
  url = {https://huggingface.co/SharleyK/predictive-maintenance-model}
}
```

## License

MIT License