Watch Demo Video
Machine Learning
Industrial Equipment Failure Prediction Using Machine Learning
The Industrial Equipment Failure Prediction Using Machine Learning system is a comprehensive machine learning solution that predicts industrial equipment failures using sensor data. This project is ideal for B.Tech, MCA, CSE, and engineering students as their final year project. It implements Lasso (L1) and Ridge (L2) Logistic Regression with leak-free SMOTE, achieving 96.6% accuracy and 0.9785 ROC-AUC.
The system utilizes the Predictive Maintenance Dataset containing 10,000 equipment records with 9 key features including temperature measurements, rotational speed, torque, and tool wear. It provides a user-friendly web interface for data upload, exploratory data analysis, model training, and real-time equipment failure prediction, enabling proactive maintenance scheduling and reduced industrial downtime.
Python 3.8+
Machine Learning
Lasso Logistic Regression
Ridge Logistic Regression
Scikit-learn
SMOTE
Pandas
NumPy
Matplotlib
Seaborn
Flask
HTML/CSS/JS
Key Features:
- Multi-Model Failure Prediction
- Ridge Logistic (96.6% Accuracy)
- Lasso Logistic (96.2% Accuracy)
- Feature Engineering (Temperature Difference, Power, Risk Score)
- Leak-free SMOTE Cross-Validation
- Interactive Visualizations
- Real-time Failure Predictions
- Model Performance Comparison
- Feature Importance Analysis
- Flask Web Application
Algorithms Used
📊 Lasso Logistic (L1)
L1-regularized logistic regression with automatic feature selection, saga solver, C=1.0
🎯 Accuracy: 96.2%
📈 Ridge Logistic (L2)
L2-regularized logistic regression with stable coefficients, lbfgs solver, C=1.0
🎯 Accuracy: 96.6%
🔧 Feature Engineering
Temperature_Difference, Power, Failure_Risk_Score, Tool_Wear_Category
📊 Top Feature: Tool Wear
🔄 SMOTE Oversampling
Synthetic minority oversampling applied leak-free inside CV folds
📊 CV AUC: 0.9763
Methodology & Workflow
1
Data Collection
10,000 equipment records with 9 sensor features
2
Data Preprocessing
Cleaning, encoding, feature engineering, scaling
3
Exploratory Data Analysis
Statistical analysis and visualization of patterns
4
Model Training
Lasso and Ridge Logistic Regression with leak-free SMOTE
5
Model Evaluation
Accuracy, Precision, Recall, F1-Score, ROC-AUC
6
Web Deployment
Flask web app with real-time failure prediction
Model Performance Comparison
| Metric |
Lasso Logistic |
Ridge Logistic |
Best |
| Test Accuracy |
96.2% |
96.6% |
Ridge |
| Test Precision |
95.9% |
96.2% |
Ridge |
| Test Recall |
96.2% |
96.6% |
Ridge |
| Test F1-Score |
96.1% |
96.4% |
Ridge |
| Test ROC-AUC |
97.7% |
97.9% |
Ridge |
| CV Mean ROC-AUC |
97.0% |
97.6% |
Ridge |
Project Package Includes:
Complete Source Code
Documentation (50+ pages)
Video Tutorial
Predictive Maintenance Dataset
Flask Web App
Model Files (Pickle)
Visualizations
24/7 Expert Support