()
| 86 | |
| 87 | |
| 88 | def test(): |
| 89 | model.eval() |
| 90 | output = model(features, adj) |
| 91 | loss_test = F.nll_loss(output[idx_test], labels[idx_test]) |
| 92 | acc_test = accuracy(output[idx_test], labels[idx_test]) |
| 93 | print("Test set results:", |
| 94 | "loss= {:.4f}".format(loss_test.item()), |
| 95 | "accuracy= {:.4f}".format(acc_test.item())) |
| 96 | |
| 97 | |
| 98 | # Train model |