Closed lcqsigi closed 2 years ago
batch_scores = model(batch_graphs, batch_X, batch_E, batch_snorm_n, batch_snorm_e) print(batch_scores.size())
batch_labels = batch_labels print(f'accuracy: {accuracy(batch_scores, batch_labels)}')
torch.Size([10, 8]) accuracy: 10
Check out this pull request on
See visual diffs & provide feedback on Jupyter Notebooks.
Powered by ReviewNB
Thanks.
batch_scores = model(batch_graphs, batch_X, batch_E, batch_snorm_n, batch_snorm_e) print(batch_scores.size())
batch_labels = batch_labels print(f'accuracy: {accuracy(batch_scores, batch_labels)}')
torch.Size([10, 8]) accuracy: 10