About prediction method

[P260]

Hi,
I was wondering if there is any predict method available for this model? I want to get classification predictions for a small set of unknown molecules I have for my downstream finetuned task using this pretrain-gnn model. Can I save the model and run predictions? I couldn't find it in here but if I am missing anything, please help me out.

Thank you!

[weihua916]

Hi! We do not offer functionality for saving the finetuned model, so you will need to write yourself.