PytorchGeometricTutorial icon indicating copy to clipboard operation
PytorchGeometricTutorial copied to clipboard

Tutorial 11 -- loss function of Node2Vec class

Open xianhong opened this issue 2 years ago • 0 comments

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 + neg_loss * self.num negative_samples" instead?

xianhong avatar Feb 04 '23 01:02 xianhong