Issue #8 Added a function to visualize the confusion matrix

[RohanHBTU]

Linked Issue

Closes: issue #8 Add a function to visualize the confusion matrix

Describe the changes you've made

Give a clear description what modifications you have made

I have added a function to visualize confusion matrix in a heatmap form with coolwarm colors. Additionally, accuracy, precision, recall and f1 score has also been added to the heatmap. Also, I imported f1 score which fixed a bug for metrics function.

Type of change

What sort of change have you made:

• [x] Bug fix (non-breaking change which fixes an issue)
• [x] New feature (non-breaking change which adds functionality)
• [ ] Code style update (formatting, local variables)
• [ ] Breaking change (fix or feature that would cause existing functionality to not work as expected)
• [ ] This change requires a documentation update

How Has This Been Tested?

Describe how have you verified the changes made

I ran some cells to show the confusion matrix and it worked perfectly without giving any error in whole notebook.

Checklist:

• [x] My code follows the guidelines of this project.
• [x] I have performed a self-review of my own code.
• [ ] I have commented my code, particularly wherever it was hard to understand.
• [ ] I have made corresponding changes to the documentation.
• [x] My changes generate no new warnings.
• [x] I have added tests that prove my fix is effective or that my feature works.
• [ ] Any dependent changes have been merged and published in downstream modules.

Screenshots

Original | Updated :: |::

Original Screenshot | Updated Screenshot |

[KrishnenduDakshi2002]

@RohanHBTU can you create another pull request with updated notebook. as your commit doesn't contain the previous code for Random Forest Classification