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

GEM pretrain model. #253

Open
karthikjetty opened this issue Mar 9, 2023 · 1 comment
Open

GEM pretrain model. #253

karthikjetty opened this issue Mar 9, 2023 · 1 comment

Comments

@karthikjetty
Copy link

Hi! I was looking through the installation guide for ChemRL GEM at the following link. https://github.com/PaddlePaddle/PaddleHelix/tree/dev/apps/pretrained_compound/ChemRL/GEM.

I have downloaded the pretrained model trained on the ZINC dataset and I notice that it says "Also, the pretrained model can be used for other molecular property prediction tasks." I was wondering what tasks it would be effective at predicting, given that the demo ZINC dataset doesn't provide any task names for the model to be trained on.

For example, would this model be accurate at predicting any of the tasks mentioned in the other datasets that could be used for further pretraining of the model?

@Noisyntrain
Copy link
Collaborator

Hi karthikjetty,
The pretrained model on ZINC can be further used to predict downstream datasets' tasks once added proper head(just like the way we use it in finetune_class.py and finetune_regr.py). Hope this can be helpful to you.

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