Fixed confusion matrix normalization issue.

MedAymenF · MedAymenF · commit 1c6cf83f87b4 · 2023-06-14T22:09:28.000Z
diff --git a/site/en/guide/core/logistic_regression_core.ipynb b/site/en/guide/core/logistic_regression_core.ipynb
@@ -1,6 +1,7 @@
 {
   "cells": [
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "FhGuhbZ6M5tl"
@@ -32,6 +33,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "EIdT9iu_Z4Rb"
@@ -41,6 +43,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "bBIlTPscrIT9"
@@ -63,6 +66,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "DauaqJ7WhIhO"
@@ -74,6 +78,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "nchsZfwEVtVs"
@@ -121,6 +126,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "gFh9ne3FZ-On"
@@ -151,6 +157,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "A3VR1aTP92nV"
@@ -182,6 +189,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "0_Z1V6Dg-La_"
@@ -202,6 +210,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "s4-Wn2jzVC1W"
@@ -268,6 +277,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "3MWuJTKEDM-f"
@@ -296,6 +306,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "J4ubs136WLNp"
@@ -316,6 +327,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "5YOG5iKYKW_3"
@@ -338,6 +350,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "_8pDCIFjMla8"
@@ -373,6 +386,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "6o3CrycBXA2s"
@@ -430,6 +444,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "VMXEhrZuKECV"
@@ -464,6 +479,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "Q_mutLj0KNUb"
@@ -489,6 +505,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "uTCndUecKZho"
@@ -528,6 +545,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "eObQu9fDnXGL"
@@ -560,6 +578,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "PribnwDHUksC"
@@ -590,6 +609,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "J_0KHQ25_2dF"
@@ -616,6 +636,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "sLiWZZPBSDip"
@@ -678,6 +699,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "NoLiAg7fYft7"
@@ -745,6 +767,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "yrj1TbOJasjA"
@@ -769,7 +792,7 @@
         "  # Compute the confusion matrix and normalize it\n",
         "  plt.figure(figsize=(10,10))\n",
         "  confusion = sk_metrics.confusion_matrix(y.numpy(), y_classes.numpy())\n",
-        "  confusion_normalized = confusion / confusion.sum(axis=1)\n",
+        "  confusion_normalized = confusion / confusion.sum(axis=1, keepdims=True)\n",
         "  axis_labels = range(2)\n",
         "  ax = sns.heatmap(\n",
         "      confusion_normalized, xticklabels=axis_labels, yticklabels=axis_labels,\n",
@@ -805,6 +828,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "DlivxaDmTnGq"
@@ -816,6 +840,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "7ADEN2rb4Nhj"
@@ -867,6 +892,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "gtofGIBN_qFd"
@@ -902,6 +928,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "vgGQuV-yqYZH"

Original file line number	Diff line number	Diff line change
`@@ -1,6 +1,7 @@`
`1`	`1`	`{`
`2`	`2`	`"cells": [`
`3`	`3`	`{`
	`4`	`+ "attachments": {},`
`4`	`5`	`"cell_type": "markdown",`
`5`	`6`	`"metadata": {`
`6`	`7`	`"id": "FhGuhbZ6M5tl"`
`@@ -32,6 +33,7 @@`
`32`	`33`	`]`
`33`	`34`	`},`
`34`	`35`	`{`
	`36`	`+ "attachments": {},`
`35`	`37`	`"cell_type": "markdown",`
`36`	`38`	`"metadata": {`
`37`	`39`	`"id": "EIdT9iu_Z4Rb"`
`@@ -41,6 +43,7 @@`
`41`	`43`	`]`
`42`	`44`	`},`
`43`	`45`	`{`
	`46`	`+ "attachments": {},`
`44`	`47`	`"cell_type": "markdown",`
`45`	`48`	`"metadata": {`
`46`	`49`	`"id": "bBIlTPscrIT9"`
`@@ -63,6 +66,7 @@`
`63`	`66`	`]`
`64`	`67`	`},`
`65`	`68`	`{`
	`69`	`+ "attachments": {},`
`66`	`70`	`"cell_type": "markdown",`
`67`	`71`	`"metadata": {`
`68`	`72`	`"id": "DauaqJ7WhIhO"`
`@@ -74,6 +78,7 @@`
`74`	`78`	`]`
`75`	`79`	`},`
`76`	`80`	`{`
	`81`	`+ "attachments": {},`
`77`	`82`	`"cell_type": "markdown",`
`78`	`83`	`"metadata": {`
`79`	`84`	`"id": "nchsZfwEVtVs"`
`@@ -121,6 +126,7 @@`
`121`	`126`	`]`
`122`	`127`	`},`
`123`	`128`	`{`
	`129`	`+ "attachments": {},`
`124`	`130`	`"cell_type": "markdown",`
`125`	`131`	`"metadata": {`
`126`	`132`	`"id": "gFh9ne3FZ-On"`
`@@ -151,6 +157,7 @@`
`151`	`157`	`]`
`152`	`158`	`},`
`153`	`159`	`{`
	`160`	`+ "attachments": {},`
`154`	`161`	`"cell_type": "markdown",`
`155`	`162`	`"metadata": {`
`156`	`163`	`"id": "A3VR1aTP92nV"`
`@@ -182,6 +189,7 @@`
`182`	`189`	`]`
`183`	`190`	`},`
`184`	`191`	`{`
	`192`	`+ "attachments": {},`
`185`	`193`	`"cell_type": "markdown",`
`186`	`194`	`"metadata": {`
`187`	`195`	`"id": "0_Z1V6Dg-La_"`
`@@ -202,6 +210,7 @@`
`202`	`210`	`]`
`203`	`211`	`},`
`204`	`212`	`{`
	`213`	`+ "attachments": {},`
`205`	`214`	`"cell_type": "markdown",`
`206`	`215`	`"metadata": {`
`207`	`216`	`"id": "s4-Wn2jzVC1W"`
`@@ -268,6 +277,7 @@`
`268`	`277`	`]`
`269`	`278`	`},`
`270`	`279`	`{`
	`280`	`+ "attachments": {},`
`271`	`281`	`"cell_type": "markdown",`
`272`	`282`	`"metadata": {`
`273`	`283`	`"id": "3MWuJTKEDM-f"`
`@@ -296,6 +306,7 @@`
`296`	`306`	`]`
`297`	`307`	`},`
`298`	`308`	`{`
	`309`	`+ "attachments": {},`
`299`	`310`	`"cell_type": "markdown",`
`300`	`311`	`"metadata": {`
`301`	`312`	`"id": "J4ubs136WLNp"`
`@@ -316,6 +327,7 @@`
`316`	`327`	`]`
`317`	`328`	`},`
`318`	`329`	`{`
	`330`	`+ "attachments": {},`
`319`	`331`	`"cell_type": "markdown",`
`320`	`332`	`"metadata": {`
`321`	`333`	`"id": "5YOG5iKYKW_3"`
`@@ -338,6 +350,7 @@`
`338`	`350`	`]`
`339`	`351`	`},`
`340`	`352`	`{`
	`353`	`+ "attachments": {},`
`341`	`354`	`"cell_type": "markdown",`
`342`	`355`	`"metadata": {`
`343`	`356`	`"id": "_8pDCIFjMla8"`
`@@ -373,6 +386,7 @@`
`373`	`386`	`]`
`374`	`387`	`},`
`375`	`388`	`{`
	`389`	`+ "attachments": {},`
`376`	`390`	`"cell_type": "markdown",`
`377`	`391`	`"metadata": {`
`378`	`392`	`"id": "6o3CrycBXA2s"`
`@@ -430,6 +444,7 @@`
`430`	`444`	`]`
`431`	`445`	`},`
`432`	`446`	`{`
	`447`	`+ "attachments": {},`
`433`	`448`	`"cell_type": "markdown",`
`434`	`449`	`"metadata": {`
`435`	`450`	`"id": "VMXEhrZuKECV"`
`@@ -464,6 +479,7 @@`
`464`	`479`	`]`
`465`	`480`	`},`
`466`	`481`	`{`
	`482`	`+ "attachments": {},`
`467`	`483`	`"cell_type": "markdown",`
`468`	`484`	`"metadata": {`
`469`	`485`	`"id": "Q_mutLj0KNUb"`
`@@ -489,6 +505,7 @@`
`489`	`505`	`]`
`490`	`506`	`},`
`491`	`507`	`{`
	`508`	`+ "attachments": {},`
`492`	`509`	`"cell_type": "markdown",`
`493`	`510`	`"metadata": {`
`494`	`511`	`"id": "uTCndUecKZho"`
`@@ -528,6 +545,7 @@`
`528`	`545`	`]`
`529`	`546`	`},`
`530`	`547`	`{`
	`548`	`+ "attachments": {},`
`531`	`549`	`"cell_type": "markdown",`
`532`	`550`	`"metadata": {`
`533`	`551`	`"id": "eObQu9fDnXGL"`
`@@ -560,6 +578,7 @@`
`560`	`578`	`]`
`561`	`579`	`},`
`562`	`580`	`{`
	`581`	`+ "attachments": {},`
`563`	`582`	`"cell_type": "markdown",`
`564`	`583`	`"metadata": {`
`565`	`584`	`"id": "PribnwDHUksC"`
`@@ -590,6 +609,7 @@`
`590`	`609`	`]`
`591`	`610`	`},`
`592`	`611`	`{`
	`612`	`+ "attachments": {},`
`593`	`613`	`"cell_type": "markdown",`
`594`	`614`	`"metadata": {`
`595`	`615`	`"id": "J_0KHQ25_2dF"`
`@@ -616,6 +636,7 @@`
`616`	`636`	`]`
`617`	`637`	`},`
`618`	`638`	`{`
	`639`	`+ "attachments": {},`
`619`	`640`	`"cell_type": "markdown",`
`620`	`641`	`"metadata": {`
`621`	`642`	`"id": "sLiWZZPBSDip"`
`@@ -678,6 +699,7 @@`
`678`	`699`	`]`
`679`	`700`	`},`
`680`	`701`	`{`
	`702`	`+ "attachments": {},`
`681`	`703`	`"cell_type": "markdown",`
`682`	`704`	`"metadata": {`
`683`	`705`	`"id": "NoLiAg7fYft7"`
`@@ -745,6 +767,7 @@`
`745`	`767`	`]`
`746`	`768`	`},`
`747`	`769`	`{`
	`770`	`+ "attachments": {},`
`748`	`771`	`"cell_type": "markdown",`
`749`	`772`	`"metadata": {`
`750`	`773`	`"id": "yrj1TbOJasjA"`
`@@ -769,7 +792,7 @@`
`769`	`792`	`" # Compute the confusion matrix and normalize it\n",`
`770`	`793`	`" plt.figure(figsize=(10,10))\n",`
`771`	`794`	`" confusion = sk_metrics.confusion_matrix(y.numpy(), y_classes.numpy())\n",`
`772`		`- " confusion_normalized = confusion / confusion.sum(axis=1)\n",`
	`795`	`+ " confusion_normalized = confusion / confusion.sum(axis=1, keepdims=True)\n",`
`773`	`796`	`" axis_labels = range(2)\n",`
`774`	`797`	`" ax = sns.heatmap(\n",`
`775`	`798`	`" confusion_normalized, xticklabels=axis_labels, yticklabels=axis_labels,\n",`
`@@ -805,6 +828,7 @@`
`805`	`828`	`]`
`806`	`829`	`},`
`807`	`830`	`{`
	`831`	`+ "attachments": {},`
`808`	`832`	`"cell_type": "markdown",`
`809`	`833`	`"metadata": {`
`810`	`834`	`"id": "DlivxaDmTnGq"`
`@@ -816,6 +840,7 @@`
`816`	`840`	`]`
`817`	`841`	`},`
`818`	`842`	`{`
	`843`	`+ "attachments": {},`
`819`	`844`	`"cell_type": "markdown",`
`820`	`845`	`"metadata": {`
`821`	`846`	`"id": "7ADEN2rb4Nhj"`
`@@ -867,6 +892,7 @@`
`867`	`892`	`]`
`868`	`893`	`},`
`869`	`894`	`{`
	`895`	`+ "attachments": {},`
`870`	`896`	`"cell_type": "markdown",`
`871`	`897`	`"metadata": {`
`872`	`898`	`"id": "gtofGIBN_qFd"`
`@@ -902,6 +928,7 @@`
`902`	`928`	`]`
`903`	`929`	`},`
`904`	`930`	`{`
	`931`	`+ "attachments": {},`
`905`	`932`	`"cell_type": "markdown",`
`906`	`933`	`"metadata": {`
`907`	`934`	`"id": "vgGQuV-yqYZH"`