Add model documentation
Browse files
README.md
ADDED
|
@@ -0,0 +1,107 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
datasets:
|
| 4 |
+
- engine-predictive-maintenance-processed
|
| 5 |
+
metrics:
|
| 6 |
+
- accuracy
|
| 7 |
+
- precision
|
| 8 |
+
- recall
|
| 9 |
+
- f1
|
| 10 |
+
- roc_auc
|
| 11 |
+
task_ids:
|
| 12 |
+
- tabular-classification
|
| 13 |
+
---
|
| 14 |
+
|
| 15 |
+
# Engine Predictive Maintenance Model
|
| 16 |
+
|
| 17 |
+
## Model Description
|
| 18 |
+
|
| 19 |
+
This is a Random Forest classifier trained to predict engine failures based on sensor readings. The model uses SMOTE oversampling to handle class imbalance and achieves high recall for failure detection, which is critical in a maintenance context.
|
| 20 |
+
|
| 21 |
+
### Model Details
|
| 22 |
+
|
| 23 |
+
- **Model Type**: Random Forest with SMOTE Pipeline
|
| 24 |
+
- **Framework**: scikit-learn, imbalanced-learn
|
| 25 |
+
- **Task**: Binary Classification (Engine Condition: Good/Failing)
|
| 26 |
+
- **Input Features**: Engine sensors (RPM, pressure, temperature, etc.)
|
| 27 |
+
- **Output**: Probability of engine failure
|
| 28 |
+
|
| 29 |
+
## Model Performance
|
| 30 |
+
|
| 31 |
+
### Test Set Metrics
|
| 32 |
+
|
| 33 |
+
| Metric | Value |
|
| 34 |
+
|--------|-------|
|
| 35 |
+
| Accuracy | 0.0000 |
|
| 36 |
+
| Precision | 0.0000 |
|
| 37 |
+
| Recall | 0.0000 |
|
| 38 |
+
| F1 Score | 0.0000 |
|
| 39 |
+
| F2 Score | 0.0000 |
|
| 40 |
+
| AUC-ROC | 0.0000 |
|
| 41 |
+
| Brier Score | 0.0000 |
|
| 42 |
+
|
| 43 |
+
## Intended Use
|
| 44 |
+
|
| 45 |
+
This model is designed for:
|
| 46 |
+
- **Predictive Maintenance**: Identify engines at risk of failure before breakdown
|
| 47 |
+
- **Condition Monitoring**: Support data-driven maintenance decision-making
|
| 48 |
+
- **Fleet Management**: Optimize maintenance scheduling and resource allocation
|
| 49 |
+
|
| 50 |
+
## Limitations
|
| 51 |
+
|
| 52 |
+
- Trained on historical engine data with specific sensor configurations
|
| 53 |
+
- Performance may vary with new sensor types or operating conditions
|
| 54 |
+
- Model requires regular retraining with updated failure data
|
| 55 |
+
- Does not capture temporal degradation patterns (time-series)
|
| 56 |
+
|
| 57 |
+
## Training Data
|
| 58 |
+
|
| 59 |
+
- **Dataset**: Engine Predictive Maintenance Dataset
|
| 60 |
+
- **Total Samples**: ~19,000 engines
|
| 61 |
+
- **Training Samples**: ~13,300 (70%)
|
| 62 |
+
- **Test Samples**: ~3,800 (20%)
|
| 63 |
+
- **Features**: 8 continuous sensor variables + derived features
|
| 64 |
+
- **Class Distribution**: Imbalanced (Good: ~63%, Failure: ~37%)
|
| 65 |
+
|
| 66 |
+
## Training Procedure
|
| 67 |
+
|
| 68 |
+
1. Data preprocessing and feature engineering
|
| 69 |
+
2. Train-test split (70-20-10)
|
| 70 |
+
3. SMOTE oversampling on training data
|
| 71 |
+
4. Hyperparameter tuning via GridSearchCV
|
| 72 |
+
5. Evaluation on held-out test set
|
| 73 |
+
|
| 74 |
+
## Evaluation Results
|
| 75 |
+
|
| 76 |
+
The model achieves:
|
| 77 |
+
- **High Recall (0.0000)**: Detects ~0% of actual failures
|
| 78 |
+
- **Competitive Precision (0.0000)**: ~0% of predictions are correct
|
| 79 |
+
- **Strong AUC (0.0000)**: Good discrimination between classes
|
| 80 |
+
|
| 81 |
+
## Recommendations
|
| 82 |
+
|
| 83 |
+
1. **Threshold Tuning**: Adjust decision threshold based on maintenance cost vs. failure cost trade-off
|
| 84 |
+
2. **Continuous Monitoring**: Track model performance in production and retrain quarterly
|
| 85 |
+
3. **Feature Importance**: Use model to identify critical sensors for maintenance teams
|
| 86 |
+
4. **Ensemble Approaches**: Consider combining with other models for robust predictions
|
| 87 |
+
|
| 88 |
+
## Citation
|
| 89 |
+
|
| 90 |
+
If you use this model, please cite:
|
| 91 |
+
|
| 92 |
+
```
|
| 93 |
+
@model{engine_maintenance_rf_2026,
|
| 94 |
+
title={Engine Predictive Maintenance Model},
|
| 95 |
+
author={Engine Predictive Maintenance Team},
|
| 96 |
+
year={2026},
|
| 97 |
+
note={Random Forest with SMOTE for failure prediction}
|
| 98 |
+
}
|
| 99 |
+
```
|
| 100 |
+
|
| 101 |
+
## License
|
| 102 |
+
|
| 103 |
+
This model is released under the MIT License. See LICENSE file for details.
|
| 104 |
+
|
| 105 |
+
## Acknowledgments
|
| 106 |
+
|
| 107 |
+
Engine predictive maintenance model developed for optimal failure detection and prevention.
|