awslabs / dgl-lifesci

Python package for graph neural networks in chemistry and biology
Apache License 2.0
714 stars 147 forks source link

Example showing how to use an ML model to tune pretrained generative models? #219

Closed PatWalters closed 1 year ago

PatWalters commented 1 year ago

Is there an example showing how one can use an ML model to tune the output of the pre-trained generative models?

mufeili commented 1 year ago

What's your purpose of tuning a pre-trained generative model?

PatWalters commented 1 year ago

I want to fine-tune the pre-trained model to output molecules meeting a specific profile. For instance, outputting molecules that score highly using a target-specific regression model.

mufeili commented 1 year ago

I see. DGL-LifeSci does not support that. You may want to check the original codebase of JTVAE or other generative models.

PatWalters commented 1 year ago

Thanks for your quick response. I'm just curious about the use case for the pre-trained generative models.

mufeili commented 1 year ago

No problems! I'll close the issue for now. Feel free to reopen it if you have any further questions.