Tabular Classification
Scikit-learn
Joblib
English
gradient-boosting
predictive-maintenance
scikit-learn
Instructions to use simnid/predictive-maintenance-model with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Scikit-learn
How to use simnid/predictive-maintenance-model with Scikit-learn:
from huggingface_hub import hf_hub_download import joblib model = joblib.load( hf_hub_download("simnid/predictive-maintenance-model", "sklearn_model.joblib") ) # only load pickle files from sources you trust # read more about it here https://skops.readthedocs.io/en/stable/persistence.html - Notebooks
- Google Colab
- Kaggle
Create README.md
Browse files
README.md
ADDED
|
@@ -0,0 +1,30 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Predictive Maintenance – Gradient Boosting Model
|
| 2 |
+
|
| 3 |
+
## Model Overview
|
| 4 |
+
This model is a recall-optimized Gradient Boosting classifier developed to support predictive maintenance for engine systems. The primary objective is to identify engines likely to require maintenance before failure occurs.
|
| 5 |
+
|
| 6 |
+
## Training Data
|
| 7 |
+
The model was trained on a prepared engine sensor dataset sourced from the Hugging Face Dataset Hub. The dataset contains structured numeric sensor readings representing engine operating conditions.
|
| 8 |
+
|
| 9 |
+
## Objective
|
| 10 |
+
- Minimize missed engine failures (false negatives)
|
| 11 |
+
- Prioritize recall for the faulty engine class
|
| 12 |
+
|
| 13 |
+
## Evaluation Metrics
|
| 14 |
+
- Recall (Faulty): ~0.84
|
| 15 |
+
- ROC-AUC: ~0.70
|
| 16 |
+
- PR-AUC: ~0.80
|
| 17 |
+
|
| 18 |
+
## Intended Use
|
| 19 |
+
This model is intended for:
|
| 20 |
+
- Predictive maintenance decision support
|
| 21 |
+
- Risk-based maintenance scheduling
|
| 22 |
+
- Offline or batch inference scenarios
|
| 23 |
+
|
| 24 |
+
## Limitations
|
| 25 |
+
- Trained on a static, pre-processed dataset
|
| 26 |
+
- Does not incorporate temporal or sequential dependencies
|
| 27 |
+
- Threshold selection may require calibration based on operational risk tolerance
|
| 28 |
+
|
| 29 |
+
## Model Artifacts
|
| 30 |
+
The repository contains a serialized `joblib` model file that can be loaded directly for inference in Python-based environments.
|