Skip to content

why the accuracy in pubmed is so high? #157

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?

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

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