Function evaluate_step

examples/graphbolt/disk_based_feature/node_classification.py:271–276

(minibatch, model)

def evaluate_step(minibatch, model):
    node_features = minibatch.node_features["feat"]
    labels = minibatch.labels
    out = model(minibatch.blocks, node_features)
    num_correct = accuracy(out, labels) * labels.size(0)
    return num_correct, labels.size(0)


@torch.no_grad()