Update README.md

inoue0426 · Nov 7, 2024 · ed98534 · ed98534
1 parent 0ed6301
commit ed98534
Showing 1 changed file with 3 additions and 6 deletions.
diff --git 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