Skip to content

why the accuracy in pubmed is so high? #157

@dkhonker

Description

@dkhonker

The code (based on docs):

from deeprobust.graph.data import Dataset
from deeprobust.graph.defense import GCN
data = Dataset(root='/tmp/', name='pubmed')
adj, features, labels = data.adj, data.features, data.labels
idx_train, idx_val, idx_test = data.idx_train, data.idx_val, data.idx_test
gcn = GCN(nfeat=features.shape[1],
          nhid=16,
          nclass=labels.max().item() + 1,
          dropout=0.5, device='cpu')
gcn = gcn.to('cpu')
gcn.fit(features, adj, labels, idx_train) # train without earlystopping
gcn.fit(features, adj, labels, idx_train, idx_val, patience=30) # train with earlystopping
gcn.test(idx_test)

And i get the test accuracy is ~86%.

But from paperswithcode and GCN paper, the accuracy of gcn is ~80%.

Do you know what caused this?

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