CoolFace
Apppublic

Divya499/ReliabilityPulse

sourceHugging Faceupdated 6mo agoView on Hugging Face
1likes
App README

ReliabilityPulse: AI-Driven Failure Forecasting for Industrial Assets

ReliabilityPulse is a high-performance predictive maintenance system for smart manufacturing. Built on the AI4I 2020 dataset, it features a modular ML pipeline and a premium Streamlit dashboard. Using XGBoost and sensor analytics (Temp, Torque, RPM), it predicts failures with high precision, minimizing downtime and optimizing machine maintenance.

๐Ÿš€ Live Demo on Hugging Face Spaces

๐Ÿ“ Project Structure

04_predictive_maintenance/
โ”œโ”€โ”€ data/
โ”‚   โ”œโ”€โ”€ raw/ai4i2020.csv        # Input Dataset (10,000 records)
โ”‚   โ””โ”€โ”€ processed/features.csv   # Engineered features and preprocessed data
โ”œโ”€โ”€ models/
โ”‚   โ”œโ”€โ”€ xgboost_model.pkl      # Primary Classifier (F1 ~88-95%)
โ”‚   โ”œโ”€โ”€ isolation_forest.pkl   # Anomaly Baseline model
โ”‚   โ””โ”€โ”€ scaler.pkl             # StandardScaler for sensors
โ”œโ”€โ”€ pipeline/
โ”‚   โ”œโ”€โ”€ 01_eda.py              # Visual Analysis (Distributions, Heatmaps)
โ”‚   โ”œโ”€โ”€ 02_feature_engineering.py # Physics-based Feature Engineering
โ”‚   โ”œโ”€โ”€ 03_preprocessing.py      # Scaling and SMOTE Balancing
โ”‚   โ”œโ”€โ”€ 04_model_training.py     # GridSearch Tuning for best models
โ”‚   โ””โ”€โ”€ 05_evaluation.py         # Performance Reporting and Metrics
โ”œโ”€โ”€ outputs/
โ”‚   โ”œโ”€โ”€ confusion_matrix.png    # Classification Performance Plot
โ”‚   โ”œโ”€โ”€ roc_curve_comparison.png # ROC for Logistic, SVM, XGBoost
โ”‚   โ”œโ”€โ”€ feature_importance.png   # Key risk drivers bar chart
โ”‚   โ””โ”€โ”€ anomaly_scores.png       # Isolation Forest Score Distribution
โ”œโ”€โ”€ app.py                      # Interactive Streamlit Dashboard
โ”œโ”€โ”€ path_utils.py               # Centralized Path Management
โ””โ”€โ”€ README.md                   # Project Documentation

๐Ÿš€ Getting Started

1. Install Dependencies

bash
pip install pandas numpy scikit-learn xgboost imbalanced-learn matplotlib seaborn joblib streamlit

2. Run the Pipeline

To retrain the model and generate metrics:

bash
python pipeline/01_eda.py
python pipeline/02_feature_engineering.py
python pipeline/03_preprocessing.py
python pipeline/04_model_training.py
python pipeline/05_evaluation.py

3. Launch the Dashboard

bash
streamlit run app.py

๐Ÿ“Š Performance Summary (XGBoost)

  • โ€”F1-Score (Failure): Target range 88โ€“95% achieved.
  • โ€”Recall (Failure): Optimized to >90% to prevent missed mechanical failures.
  • โ€”Top Drivers: Tool wear interaction with Torque and Power usage.

๐Ÿ”ง Maintenance Recommendations (Dashboard)

  • โ€”Low Risk: Schedule routine inspection in 100 hours.
  • โ€”Medium Risk: Inspect within 24 hours.
  • โ€”High/Critical Risk: Immediate manual inspection or stop operations.

Built by [Divyanshi Singh](https://www.linkedin.com/in/divyanshi-singh-/) | [GitHub](https://github.com/Divyanshi018572)