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

CATH dataset pretraining hyper-params #12

Open
yzf-code opened this issue Dec 21, 2024 · 0 comments
Open

CATH dataset pretraining hyper-params #12

yzf-code opened this issue Dec 21, 2024 · 0 comments

Comments

@yzf-code
Copy link

Dear authors,
I have followed you description in the README to download the CATH42 datasets and processed it into protein.rball.edges.CATH42.npy, protein.knn.edges.CATH42.npy, protein.nodes.CATH42.pt.
I directly use your codes and the following commands:
python -B train.py --dataset SHS27k --split_mode random --pre_train CATH42
python -B train.py --dataset SHS27k --split_mode dfs --pre_train CATH42
python -B train.py --dataset SHS27k --split_mode bfs --pre_train CATH42

But sadly I found that my test results are much lower than those reported in your paper.

I think this may be caused by the inconsistency between the hyperparameters for pretraining with CATH42 and the hyperparameters without CATH42 which is provided in your code. Could you please provide the hyperparameters for pretraining with CATH42?
Thank you very much!

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

1 participant