Skip to content

GitLab

Unverified Commit 6d346281 authored by Zihao Ye's avatar Zihao Ye Committed by GitHub
Browse files

upd (#2043)

parent 1a1f38e3
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
examples/pytorch/gat/train_ppi.py
View file @ 6d346281
...@@ -119,7 +119,8 @@ def main(args): ...@@ -119,7 +119,8 @@ def main(args):
test_score_list = [] test_score_list = []
for batch, subgraph in enumerate(test_dataloader): for batch, subgraph in enumerate(test_dataloader):
subgraph = subgraph.to(device) 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())) print("Test F1-Score: {:.4f}".format(np.array(test_score_list).mean()))
if __name__ == '__main__': if __name__ == '__main__':
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment