return prob

inoue0426 · Nov 7, 2024 · 4c599ab · 4c599ab
1 parent 15f1fe9
commit 4c599ab
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/drGAT.py b/drGAT.py
@@ -1,5 +1,5 @@
 """
-This is the official implementation of "drGAT: Attention-Guided Gene Assessment 
+This is the official implementation of "drGAT: Attention-Guided Gene Assessment
 for Drug Response in Drug-Cell-Gene Heterogeneous Network."
 
 Written by inoue0426
@@ -303,4 +303,4 @@ def eval(model, data, device=None):
         test_labels.cpu().detach().numpy(), predict.cpu().detach().numpy()
     )
 
-    return res
+    return predict, res