Closed Arihant-Bhandari closed 2 months ago
Our team will soon review your PR. Thanks @Arihant-Bhandari :)
@abhisheks008 hi, as you suggest in the last PR, i have made appropriate changes, if there is still something that needs changes, please let me know, i would make them.
thank you for your time reviewing this PR. Hope to hear from you soon.
Pull Request for DL-Simplified đĄ
Issue Title : [Model Enhancement]: Colour Detection using DL
Closes: #529 [Model Enhancement]: Colour Detection using DL
Describe the add-ons or changes you've made đ
i have enchanced the previous model by adding the new image loading function from keras.utils, Attention mechanism, using ELU instead of ReLU due to its better functionality, made the problem into a multi-class classification model with softmax implementation, used the optimizer Nadam with lr = 0.01, displayed the first 32 kernels based on the images, saved and included the best model as part of the PR.
Type of change âď¸
What sort of change have you made:
How Has This Been Tested? âď¸
Previous Accuracy : 0.9
New metrics:
Accuracy: 0.94
Precision: 0.9565476190476192
Recall: 0.9514285714285714
F1 Score: 0.95005439005439
Confusion Matrix: [[6 0 0 0 1 0 0 0 0 0] [0 4 0 0 0 0 0 0 0 0] [0 0 7 0 0 0 0 0 0 0] [0 0 0 2 0 0 0 0 0 0] [0 0 0 0 6 0 0 0 0 0] [0 0 0 0 0 4 0 0 0 1] [0 0 1 0 0 0 6 0 0 0] [0 0 0 0 0 0 0 5 0 0] [0 0 0 0 0 0 0 0 2 0] [0 0 0 0 0 0 0 0 0 5]]
Checklist: âď¸