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
| language: | |
| - en | |
| license: apache-2.0 | |
| tags: | |
| - tabular-classification | |
| - gradient-boosting | |
| - predictive-maintenance | |
| - scikit-learn | |
| library_name: sklearn | |
| datasets: | |
| - engine-sensor-data # Replace with your actual dataset name if known | |
| metrics: | |
| - recall | |
| - roc_auc | |
| - pr_auc | |
| # Predictive Maintenance – Gradient Boosting Model | |
| ## Model Overview | |
| 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. | |
| ## Training Data | |
| 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. | |
| ## Objective | |
| - Minimize missed engine failures (false negatives) | |
| - Prioritize recall for the faulty engine class | |
| ## Evaluation Metrics | |
| - Recall (Faulty): ~0.84 | |
| - ROC-AUC: ~0.70 | |
| - PR-AUC: ~0.80 | |
| ## Intended Use | |
| This model is intended for: | |
| - Predictive maintenance decision support | |
| - Risk-based maintenance scheduling | |
| - Offline or batch inference scenarios | |
| ## Limitations | |
| - Trained on a static, pre-processed dataset | |
| - Does not incorporate temporal or sequential dependencies | |
| - Threshold selection may require calibration based on operational risk tolerance | |
| ## Model Artifacts | |
| The repository contains a serialized `joblib` model file that can be loaded directly for inference in Python-based environments. |