AntonioLonga / PytorchGeometricTutorial

Pytorch Geometric Tutorials
1.05k stars 307 forks source link

Tutorial 11 -- loss function of Node2Vec class #14

Open xianhong opened 1 year ago

xianhong commented 1 year ago

Reading the loss function of Node2Vec class (https://pytorch-geometric.readthedocs.io/en/latest/_modules/torch_geometric/nn/models/node2vec.html) , i feel the return value of "pos_loss + neg_loss" does not seem to take into account of 'num_negative_samples'. Shouldn't the loss function return "pos_loss + negloss * self.num negative_samples" instead?