use spam / not spam labels

rasbt · May 11, 2024 · 4b4e1e1 · 4b4e1e1
1 parent 45f6e72
commit 4b4e1e1
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/ch06/01_main-chapter-code/ch06.ipynb b/ch06/01_main-chapter-code/ch06.ipynb
@@ -2184,7 +2184,7 @@
     "    predicted_label = torch.argmax(logits, dim=-1).item()\n",
     "\n",
     "    # Return the classified result\n",
-    "    return \"Positive\" if predicted_label == 1 else \"Negative\""
+    "    return \"spam\" if predicted_label == 1 else \"not spam\""
    ]
   },
   {
@@ -2211,7 +2211,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Positive\n"
+      "spam\n"
      ]
     }
    ],
@@ -2242,7 +2242,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Negative\n"
+      "not spam\n"
      ]
     }
    ],
@@ -2347,7 +2347,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.12"
+   "version": "3.11.4"
   }
  },
  "nbformat": 4,