/any-domain-pretrain-gnns

Strategies for Pre-training Graph Neural Networks for Any domain

Primary LanguageJupyter NotebookMIT LicenseMIT

Strategies for Pre-training Graph Neural Networks for Any Domain

This is the any-domain implementation of Strategies for Pre-training Graph Neural Networks

Implemented pretraining methods:

  1. Edge Masking
  2. Node Masking
  3. Node and edge masking
  4. Deep Graph Infomax