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

pre-training GNN architecture on chem dataset #4

Open
wubo2180 opened this issue Jan 8, 2022 · 3 comments
Open

pre-training GNN architecture on chem dataset #4

wubo2180 opened this issue Jan 8, 2022 · 3 comments

Comments

@wubo2180
Copy link

wubo2180 commented Jan 8, 2022

Hi, do you have the pre-trained GNN model file on chem dataset

@rootlu
Copy link
Owner

rootlu commented Jan 8, 2022

Hi, we conducted experiments on the Chem dataset, but we didn't save the results and the pre-trained GNN model.
If you are interested in the performance on the Chem dataset, you can directly run the model on it.

Thx.

@wubo2180
Copy link
Author

wubo2180 commented Jan 8, 2022

Thanks. However, there are some mistakes when the GNN type is gcn, graphsage or gat (gin is OK). Error:

/model_chem.py , line 376, in forward

self_loop_attr[:, 7] = 1  # for bio dataset
IndexError: index 7 is out of bounds for dimension 1 with size 2

Should I comment out this line of code when I conduct experiment on chem dataset?

@wubo2180
Copy link
Author

wubo2180 commented Jan 8, 2022

By the way, there are some mistaks in util.py
`L2P-GNN/code/util.py", line 31, in call

i, _, k = structured_negative_sampling(data.edge_index)
NameError: name structured_negative_sampling is not defined`

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