From ed9853474cffbb2b31b87523cc1a07749965aa17 Mon Sep 17 00:00:00 2001
From: Yoshitaka Inoue <8393063+inoue0426@users.noreply.github.com>
Date: Thu, 7 Nov 2024 15:55:00 -0500
Subject: [PATCH] Update README.md

---
 README.md | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/README.md b/README.md
index 68fcafc..7ee15d4 100644
--- a/README.md
+++ b/README.md
@@ -52,20 +52,17 @@ Output is as follows:
 
 ```python
 predict, res = drGAT.eval(model, test)
-res
-```
-
+res # Metrics
 
 | Accuracy | Precision | Recall | F1 Score | True Positive | True Negative | False Positive | False Negative |
 |-----------|-----------|---------|-----------|----------------|---------------|----------------|-----------------|
 | 0.771375 | 0.740881 | 0.783245 | 0.761474 | 1178 | 1312 | 412 | 326 |
 
-```python
-predict
+predict # Probability
 
+tensor([1., 0., 0.,  ..., 1., 0., 0.])
 ```
 
-tensor([1., 0., 0.,  ..., 1., 0., 0.])
 
 ## Training