From 3cebe75fbc1a83f5a0b2fd547af049a1432b09a8 Mon Sep 17 00:00:00 2001
From: inoue0426 <8393063+inoue0426@users.noreply.github.com>
Date: Thu, 7 Nov 2024 17:03:31 -0500
Subject: [PATCH] add comments

---
 Tutorial.ipynb | 15 ++++++++-------
 1 file changed, 8 insertions(+), 7 deletions(-)

diff --git a/Tutorial.ipynb b/Tutorial.ipynb
index 7d3084f..8749766 100644
--- a/Tutorial.ipynb
+++ b/Tutorial.ipynb
@@ -168,7 +168,7 @@
     }
    ],
    "source": [
-    "predict, res = drGAT.eval(model, test)\n",
+    "predict, res = drGAT.eval(model, test) # Probability and Metrics\n",
     "res"
    ]
   },
@@ -181,7 +181,7 @@
     {
      "data": {
       "text/plain": [
-       "tensor([1., 0., 0.,  ..., 1., 0., 0.])"
+       "tensor([0.7653, 0.3292, 0.3037,  ..., 0.9121, 0.4277, 0.2037])"
       ]
      },
      "execution_count": 7,
@@ -211,7 +211,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 11,
+   "execution_count": 8,
    "id": "6097126f-f158-495b-bb1d-b8131a01ac13",
    "metadata": {},
    "outputs": [
@@ -246,7 +246,7 @@
        " tensor([1])]"
       ]
      },
-     "execution_count": 11,
+     "execution_count": 8,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -260,22 +260,23 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 12,
+   "execution_count": 9,
    "id": "986b0120-115d-4f84-bb87-ca6b3e45b34c",
    "metadata": {},
    "outputs": [
     {
      "data": {
       "text/plain": [
-       "tensor(1.)"
+       "tensor(0.7653)"
       ]
      },
-     "execution_count": 12,
+     "execution_count": 9,
      "metadata": {},
      "output_type": "execute_result"
     }
    ],
    "source": [
+    "# Probability\n",
     "predict, _ = drGAT.eval(model, test)\n",
     "predict"
    ]