-
-
Notifications
You must be signed in to change notification settings - Fork 213
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #479 from YashSachan2/yash2
Bearing Classification
- Loading branch information
Showing
9 changed files
with
3,719 additions
and
0 deletions.
There are no files selected for viewing
1,000 changes: 1,000 additions & 0 deletions
1,000
Bearing Classification/Dataset/Faulty-bearing.csv
Large diffs are not rendered by default.
Oops, something went wrong.
1,000 changes: 1,000 additions & 0 deletions
1,000
Bearing Classification/Dataset/Healthy-bearing.csv
Large diffs are not rendered by default.
Oops, something went wrong.
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
1,649 changes: 1,649 additions & 0 deletions
1,649
Bearing Classification/Model/bearing_classification.ipynb
Large diffs are not rendered by default.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,70 @@ | ||
# Bearing Classification using ML | ||
|
||
## PROJECT TITLE | ||
|
||
Bearing Classification | ||
|
||
## GOAL | ||
|
||
To identify faulty and healthy bearing. | ||
|
||
## DATASET | ||
|
||
The link for the dataset used in this project: https://www.kaggle.com/datasets/zlemglsmklkaya/healthy-vs-faulty-bearings/data?select=Healthy-bearing.csv | ||
|
||
## EDA: | ||
![Alt text](Images/Input_Dataset.png) | ||
![Alt text](Images/EDA1.png) | ||
Shape: (1998,2) | ||
|
||
## DESCRIPTION | ||
|
||
This project aims to identify the faulty and helthy bearings. | ||
|
||
## WHAT I HAD DONE | ||
|
||
1. Data collection: From the link of the dataset given above. | ||
2. Data preprocessing: Preprocessed the data to create valid features. | ||
3. Model selection: XGBC,Random Forest,Logestic Regression,Gaussian Bayes,AdaBoost Classifier. | ||
4. Comparative analysis: Compared the accuracy score of all the models. | ||
|
||
|
||
## MODELS SUMMARY | ||
|
||
- XGBC | ||
- Logistic Regression | ||
- Adaboost Classifier | ||
- Random Forest Classifier | ||
- Gaussian Bayes | ||
|
||
## LIBRARIES NEEDED | ||
|
||
The following libraries are required to run this project: | ||
|
||
- matplotlib | ||
- numpy | ||
- pandas | ||
- sklearn | ||
|
||
## EVALUATION METRICS | ||
|
||
The evaluation metrics I used to assess the models: | ||
|
||
- Accuracy | ||
|
||
It is shown using Confusion Matrix in the Images folder | ||
|
||
## RESULTS | ||
Results on Val dataset: | ||
XGBC: 77.33% | ||
Random Forest: 77.67% | ||
Adaboost: 75% | ||
Logistic Regression: 74% | ||
Gaussian Bayes: 73.33% | ||
DTC:77.33% | ||
![Alt text](Images/Metrics.png) | ||
|
||
## CONCLUSION | ||
Based on results we can draw following conclusions: | ||
|
||
1.The Random Forest worked the best |