Shypd.aiThe largest open index of AI toolspretrain-gnns is an open-source AI framework that provides PyTorch implementations for pre-training Graph Neural Networks. It offers strategies for self-supervised and supervised pre-training, enhancing GNN performance.
Trending
About
pretrain-gnns is a PyTorch implementation offering various strategies for pre-training Graph Neural Networks (GNNs). This open-source tool allows users to improve the performance of GNNs through both self-supervised methods like context prediction, masking, edge prediction, and deep graph infomax, as well as supervised pre-training. It supports fine-tuning pre-trained models on downstream datasets and provides pre-trained models for biology and chemistry applications. The project includes installation instructions, dataset download links, and scripts for reproducing paper results, making it a valuable resource for researchers and developers in graph representation learning.
Capabilities
Pricing & Plans
Open Source
Free
FAQs
Trending
