/GEARSage-DGraphFin

Primary LanguagePython

This repo provides a collection of baselines for DGraphFin dataset. Please download the dataset from the DGraph web and place it under the folder './dataset/DGraphFin/raw'.

Environments

Implementing environment:

  • numpy = 1.23.1

  • pytorch = 1.10.1

  • torch_geometric = 2.0.4

  • torch_scatter = 2.0.9

  • torch_sparse = 0.6.13

  • GPU: Tesla A100 40G

Training

  • GEARSage
python main.py

Results:

Performance on DGraphFin(10 runs):

Methods Params Valid AUC Test AUC
GEARSage 50544 0.8388 ± 0.0002 0.8460 ± 0.0002