Skip to content

storyandwine/GEARSage-DGraphFin

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages