pos_weight should be a Tensor? about gae-pytorch

[shionhonda]

I fix it by replacing this:

gae-pytorch/gae/train.py

Line 48 in 22c9edf

pos_weight = float(adj.shape[0] * adj.shape[0] - adj.sum()) / adj.sum()

to this:

pos_weight = torch.Tensor([float(adj.shape[0] * adj.shape[0] - adj.sum()) / adj.sum()])

[iamadog3333]

It works.

but I found that length of pos_weight is 1.
According documents:
pos_weight (Tensor, optional) – a weight of positive examples. Must be a vector with length equal to the number of classes.
pos_weight 's length should be 2.

[iamadog3333]

Doc says: where c is the class number (c>1 for multi-label binary classification, c=1 for single-label binary classification)
That is right.

[Dzhilin]

Doc says: where c is the class number (c>1 for multi-label binary classification, c=1 for single-label binary classification)
Hi, I have the same problem. Has your problem been solved？