How to test Yelp dataset

[LukeLIN-web]

I just use /examples/reddit.py. https://github.com/pyg-team/pytorch_geometric/blob/master/examples/reddit.py
Simply change reddit to Yelp dataset

# dataset = Reddit(path)
dataset = Yelp("/data/ogb/"+ "Yelp")

But it shows:

Traceback (most recent call last):
  File "reddit.py", line 112, in <module>
    loss, acc = train(epoch)
  File "reddit.py", line 91, in train
    total_correct += int((y_hat.argmax(dim=-1) == y).sum())
RuntimeError: The size of tensor a (1024) must match the size of tensor b (100) at non-singleton dimension 1

[rusty1s]

YELP is a multi-label classification dataset, not a multi-class classification dataset, so the equality comparison fails here.