5a9c6aa03ea55ad85601c0c33ff762d6a2e2c7fa,examples/node2vec.py,,,#,10

Before Change

path = osp.join(osp.dirname(osp.realpath(__file__)), "..", "data", dataset)
dataset = Planetoid(path, dataset)
data = dataset[0]
loader = DataLoader(torch.arange(data.num_nodes), batch_size=128, shuffle=True)

device = "cuda" if torch.cuda.is_available() else "cpu"
model = Node2Vec(data.num_nodes, embedding_dim=128, walk_length=20,

After Change

data = dataset[0]

device = "cuda" if torch.cuda.is_available() else "cpu"
model = Node2Vec(data.edge_index, embedding_dim=128, walk_length=20,
                 context_size=10, walks_per_node=10, num_negative_samples=5,
                 sparse=True).to(device)

loader = model.loader(batch_size=128, shuffle=True, num_workers=6)
optimizer = torch.optim.SparseAdam(model.parameters(), lr=0.01)

Instances