SkillAgentSearch skills...

Gipa

a general information propagation algorithm for graph learning

Install / Use

/learn @yongchao-liu/Gipa
About this skill

Quality Score

0/100

Supported Platforms

Universal

README

GIPA: General Information Propagation Algorithm

This repository is the DGL-implemention of GIPA (General Information Propagation Algorithm). The following figure illustrates the process of GIPA:

gipa_process

Datasets

The protein dataset ogbn-proteins from Open Graph Benchmark.

Usage

To train the model:

source main.sh

Or:

python src/main.py --n-layers 6 --n-epochs 8000 --lr 0.01 --batch-size 8000 --use-label --if-save --preprocess --n-hop 1 --gpu 0 --eval-every 5 --seed 0

To use the pre-trained model to do inference: see the Jupyter notebook in /example/gipa_test.ipynb.

Requirements

ogb==1.3.0
numpy==1.19.2
torch==1.7.1
dgl_cu110==0.5.3

Related Skills

View on GitHub
GitHub Stars13
CategoryEducation
Updated2y ago
Forks3

Languages

Python

Security Score

60/100

Audited on Apr 19, 2023

No findings