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
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.
- Downloads last month
- -