upd (dmlc#2043)

sanchit-misra · Aug 18, 2020 · 6d34628 · 6d34628
1 parent 1a1f38e
commit 6d34628
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/pytorch/gat/train_ppi.py b/examples/pytorch/gat/train_ppi.py
@@ -119,7 +119,8 @@ def main(args):
     test_score_list = []
     for batch, subgraph in enumerate(test_dataloader):
         subgraph = subgraph.to(device)
-        test_score_list.append(evaluate(subgraph.ndata['feat'], model, subgraph, subgraph.ndata['label'], loss_fcn))
+        score, test_loss = evaluate(subgraph.ndata['feat'], model, subgraph, subgraph.ndata['label'], loss_fcn)
+        test_score_list.append(score)
     print("Test F1-Score: {:.4f}".format(np.array(test_score_list).mean()))
 
 if __name__ == '__main__':