https://github.com/edisonleeeee/hetero2net

https://github.com/edisonleeeee/hetero2net

Last synced: about 1 year ago
JSON representation

• Host: GitHub
• URL: https://github.com/edisonleeeee/hetero2net
• Owner: EdisonLeeeee
• Created: 2023-10-13T11:06:12.000Z (over 2 years ago)
• Default Branch: master
• Last Pushed: 2024-08-18T04:53:50.000Z (almost 2 years ago)
• Last Synced: 2024-08-18T05:32:39.645Z (almost 2 years ago)
• Language: Python
• Size: 26.4 KB
• Stars: 2
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

          
# Hetero $^2$ Net

Implementation of Hetero $^2$ Net from the "Hetero $^2$ Net: Heterophily-aware Representation Learning on Heterogenerous Graphs" paper.

# Requirements

+ torch                    2.0.1

+ torch_geometric          2.4.0

+ torch-cluster            1.6.1+pt20cu117

+ torch-scatter            2.1.1+pt20cu117

+ torch-sparse             0.6.17+pt20cu117

+ torch-spline-conv        1.2.2+pt20cu117

+ texttable                1.6.7

+ termcolor                1.1.0

+ scikit-learn             1.0.2

+ scipy                    1.10.1

+ numpy                    1.22.4

+ tqdm                     4.64.0

+ CUDA 11.7

# Reproduction

## DBLP

python main.py --dataset DBLP --dropout 0.5 --lr 0.001 --beta 0.2 --alpha 0. --mask_lp --p 0.7

## IMDB

python main.py --dataset IMDB--dropout 0.9 --lr 0.005  --alpha 0.1 --mask_lp --p 0.6

## ACM

python main.py --dataset ACM --epochs 100 --mask_lp --p 0.7

## OGB-MAG

python main.py --dataset MAG --num_neighbors 15 15 --hidden 256 --beta 0.5 --mask_lp --p 1.0

## RCDD

python main.py --dataset RCDD --metrics ap micro-f1 macro-f1 --dropout 0.5 --epochs 100 --num_neighbors 15 15 --hidden 256 --beta 0.2 --mask_lp