Spaces:
Sleeping
Sleeping
| title: ReliabilityPulse | |
| emoji: ⚡ | |
| colorFrom: blue | |
| colorTo: purple | |
| sdk: streamlit | |
| app_file: app.py | |
| pinned: false | |
| # 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](https://huggingface.co/spaces/Divya499/ReliabilityPulse) | |
| ## 📁 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)** | |