Upload README.md with huggingface_hub
Browse files
README.md
ADDED
|
@@ -0,0 +1,47 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
language: en
|
| 3 |
+
tags:
|
| 4 |
+
- machine-learning
|
| 5 |
+
- classification
|
| 6 |
+
- predictive-maintenance
|
| 7 |
+
- engine-condition
|
| 8 |
+
- scikit-learn
|
| 9 |
+
datasets:
|
| 10 |
+
- dhani10/engine-condition-dataset
|
| 11 |
+
metrics:
|
| 12 |
+
- accuracy
|
| 13 |
+
- f1
|
| 14 |
+
- precision
|
| 15 |
+
- recall
|
| 16 |
+
- roc_auc
|
| 17 |
+
---
|
| 18 |
+
|
| 19 |
+
# Engine Condition Prediction Model
|
| 20 |
+
|
| 21 |
+
This model predicts engine condition (Normal=0 / Faulty=1) from sensor signals for predictive maintenance.
|
| 22 |
+
|
| 23 |
+
## Model Details
|
| 24 |
+
- **Winning Model**: Random Forest
|
| 25 |
+
- **Training Data**: dhani10/engine-condition-dataset
|
| 26 |
+
- **Input Features**: ['Engine rpm', 'Lub oil pressure', 'Fuel pressure', 'Coolant pressure', 'lub oil temp', 'Coolant temp']
|
| 27 |
+
- **Target**: Engine Condition (0=Normal, 1=Faulty)
|
| 28 |
+
- **Training Samples**: 15628
|
| 29 |
+
- **Test Samples**: 3907
|
| 30 |
+
- **Registered**: 2025-11-07T07:50:04.426743+00:00
|
| 31 |
+
|
| 32 |
+
## Cross-Validation
|
| 33 |
+
- **Best CV Score**: 0.7724
|
| 34 |
+
|
| 35 |
+
## Test Set Performance
|
| 36 |
+
- **Accuracy**: 0.6596
|
| 37 |
+
- **F1-Score**: 0.7684
|
| 38 |
+
- **Precision**: 0.6728
|
| 39 |
+
- **Recall**: 0.8957
|
| 40 |
+
- **ROC-AUC**: 0.6992
|
| 41 |
+
|
| 42 |
+
## Best Hyperparameters
|
| 43 |
+
```json
|
| 44 |
+
{
|
| 45 |
+
"classifier__max_depth": 5,
|
| 46 |
+
"classifier__n_estimators": 100
|
| 47 |
+
}
|