add comments

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"
    ]