nilanjanadevc commited on
Commit
58b9f0a
·
verified ·
1 Parent(s): 5a13c79

Add model documentation

Browse files
Files changed (1) hide show
  1. README.md +107 -0
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.