DSCKGEC / learn-machine-learn

A one-stop repository for new-comers in Machine Learning and A.I.
MIT License
1 stars 4 forks source link

Added a function to visualize confusion matrix #34

Closed RohanHBTU closed 1 year ago

RohanHBTU commented 1 year ago

Linked Issue

Closes: issue #8 Added a function visualize 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:

How Has This Been Tested?

Describe how have you verified the changes made

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

Checklist:

Screenshots

Original | Updated :Screenshot (139): |:Screenshot (137): Original Screenshot | Updated Screenshot |

anuragc2001 commented 1 year ago

@KrishnenduDakshi2002 if everything looks good and you've no issues with the PR, we may merge it.

KrishnenduDakshi2002 commented 1 year ago

@KrishnenduDakshi2002 if everything looks good and you've no issues with the PR, we may merge it.

okay I am merging PR and close this issue.