for data in test_loader: # Get classifications _, out_gcn = gcn(data.x, data.edge_index, data.batch) _, out_gin = gin(data.x, data.edge_index, data.batch) out = (out_gcn + out_gin)/2 # Calculate accuracy scores acc_gcn += accuracy(out_gcn.argmax(dim=1), data.y) / len(test_loader) acc_gin += accuracy(out_gin.argmax(dim=1), data.y) / len(test_loader) acc += accuracy(out.argmax(dim=1), data.y) / len(test_loader)