Skip to content

Commit a9283c6

Browse files
authored
Update README.md
1 parent 0e17970 commit a9283c6

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

README.md

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -143,17 +143,18 @@ This project demonstrates how domain knowledge, feature construction, and ensemb
143143

144144
###### Skills Applied: Machine Learning, Feature Engineering, SMOTE, SHAP, Ensemble Modeling, Supervised Learning, Python (scikit-learn, XGBoost, imbalanced-learn)
145145

146-
#### Figure 6: Confusion Matrix — XGBoost Model
147-
##### This confusion matrix shows the performance of the XGBoost model in detecting machine failures. It achieved 98.6% accuracy, with only 55 total misclassifications out of 3,865 samples. The model demonstrates excellent precision and recall, making it highly effective for predictive maintenance tasks in manufacturing.
148-
149-
<img src="ConfusionMatrix_XGBoost.png" width="400" />
150-
151-
#### Figure 7: SHAP Summary Plot — Feature Impact on Predictions
146+
#### Figure 6: SHAP Summary Plot — Feature Impact on Predictions
152147
##### SHAP summary plot showing how each feature influences the model’s predictions. Tool_wear_[min], Power_[W], and Rotational_speed_[rpm] are among the most influential inputs. This visualization aids in validating model behavior and supports communication with non-technical stakeholders.
153148

154149
<img src="SHAP.png" width="500" />
155150

156151

152+
#### Figure 7: Confusion Matrix — XGBoost Model
153+
##### This confusion matrix shows the performance of the XGBoost model in detecting machine failures. It achieved 98.6% accuracy, with only 55 total misclassifications out of 3,865 samples. The model demonstrates excellent precision and recall, making it highly effective for predictive maintenance tasks in manufacturing.
154+
155+
<img src="ConfusionMatrix_XGBoost.png" width="390" />
156+
157+
157158

158159
&nbsp;
159160

0 commit comments

Comments
 (0)