Skip to content

Is this an Error about the code in graph/train.py? #17

@xdjwolf

Description

@xdjwolf

when I run and debug the code in graph/train.py

Use the MUTAG dataset for example:

258 np.random.shuffle(train_idx) this line shuffled the 188 graph index
265 mask = num_nodes[idx: idx + batch_size] ,
but the line 265 to read the num_nodes as a mask, still use the original index?

And another question:

Is the mask in 265 necessaries, when in the GCN class, the forward function does not use the mask for calculate?
def forward(self, feat, adj, mask):

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions