Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

The exp result reproduce of the whole dataset on Table 2 #4

Open
gegemy opened this issue Dec 9, 2022 · 1 comment
Open

The exp result reproduce of the whole dataset on Table 2 #4

gegemy opened this issue Dec 9, 2022 · 1 comment

Comments

@gegemy
Copy link

gegemy commented Dec 9, 2022

Hi, I wonder how to produce the whole dataset's experiment results shown in Table 2. Does it directly run the whole graph (like Cora) on a model (if yes, does it run on GCN?), without any reduction rate? or set the reduction rate as 1?

@ChandlerBang
Copy link
Owner

ChandlerBang commented Jan 12, 2023

Hi,

Sorry for the late reply. This is indeed included in the train_coreset.py file:

GCond/train_coreset.py

Lines 61 to 70 in 9f2032c

# Setup GCN Model
device = 'cuda'
model = GCN(nfeat=features.shape[1], nhid=256, nclass=labels.max()+1, device=device, weight_decay=args.weight_decay)
model = model.to(device)
model.fit(features, adj, labels, idx_train, idx_val, train_iters=600, verbose=False)
model.eval()
# You can use the inner function of model to test
model.test(idx_test)

You may run train_coreset.py to get the result you want.

python train_coreset.py --dataset cora --r=0.01  --method=random
python train_coreset_induct.py --dataset flickr --r=0.01  --method=random

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants