Merge pull request #2235 from MedAymenF:bug-fix

PiperOrigin-RevId: 547622792

Author: copybara-github

site/en/guide/core/logistic_regression_core.ipynb

@@ -769,7 +769,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",