[Model Enhancement]: Colour Detection using DL #547
Closed
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
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? ⚙️
The model was tested on Accuracxy, Precision, F1-score, Recall and confusion matrtix:
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: ☑️