Skip to content

Commit

Permalink
remove unuseful comment and clarify test acc for mag240M.
Browse files Browse the repository at this point in the history
  • Loading branch information
mfbalin committed Aug 22, 2024
1 parent 9b7d71d commit 8f2f9f6
Showing 1 changed file with 1 addition and 10 deletions.
11 changes: 1 addition & 10 deletions examples/graphbolt/pyg/hetero/node_classification.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,16 +33,6 @@ def create_dataloader(
job,
):
"""Create a GraphBolt dataloader for training, validation or testing."""

###########################################################################
# Initialize the ItemSampler to sample mini-batches from the dataset.
# `item_set`:
# The set of items to sample from. This is typically the
# training, validation or test set.
# `batch_size`:
# The number of nodes to sample in each mini-batch.
# `shuffle`:
# Whether to shuffle the items in the dataset before sampling.
datapipe = gb.ItemSampler(
itemset,
batch_size=batch_size,
Expand Down Expand Up @@ -546,6 +536,7 @@ def main():
args.device,
)

# Labels are currently unavailable for mag240M so the test acc will be 0.
print("Testing...")
test_acc = evaluate(
model,
Expand Down

0 comments on commit 8f2f9f6

Please sign in to comment.