search

kimsu55 / ToxExpert

9 stars 2 forks source link

Pretrain GNN? #2

Open zhwl2117 opened 6 months ago

zhwl2117 commented 6 months ago

Dear authors,

Thank you for your interesting work. I would like to run your code in my own machine. But I cannot find which pretrained model/strategy the pretrained GIN reported in the main table is. Could you please provide more details about which pretrained model is used as the pretrained GIN?

Thanks

kimsu55 commented 6 months ago

please refer to the paper "Strategies For Pre-training Graph Neural Networks". I used "supervised_edgepred.pth" downloaded from https://github.com/snap-stanford/pretrain-gnns/tree/master/bio/model_gin.

I hope this helps you with your research.