MicrosoftDocs / ml-basics

Exercise notebooks for Machine Learning modules on Microsoft Learn

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Swapped axes on heatmap in classification notebook

cgrant42 opened this issue · comments

The axes on the heatmap in the classification notebook are incorrect. Predicted species should be on the horizontal axis and actual species should be on the vertical axis. The scikit-learn documentation for the confusion matrix states that it returns "Confusion matrix whose i-th row and j-th column entry indicates the number of samples with true label being i-th class and prediced label being j-th class."

Nice catch! Fixed throughout.