snap-stanford / GEARS

GEARS is a geometric deep learning model that predicts outcomes of novel multi-gene perturbations
MIT License
189 stars 38 forks source link

Are there pretrained weights available? #67

Closed jamesbraza closed 4 months ago

jamesbraza commented 4 months ago

GEARS.load_pretrained loads in a model.pt to GEARS.model.

Where can one find that model.pt file? Is there a pre-trained variant available somewhere (e.g. Hugging Face)?

yhr91 commented 4 months ago

Yes, there is a pretrained model available through this Colab notebook