Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
You can not select more than 25 topics
Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Dreamer
3693483479
|
1 year ago | |
|---|---|---|
| modules | 1 year ago | |
| utils | 1 year ago | |
| .DS_Store | 1 year ago | |
| README.md | 1 year ago | |
| main.py | 1 year ago | |
README.md
GraphTransformer-DGraphFin
This is the Graph Transformer solution for DGraphFin leaderboard.
Dependencies
python = 3.8
torch = 1.10.1+cu102
torch-geometric = 2.0.4
scikit-learn = 0.20.4
torch-scatter = 2.0.9
torch-sparse = 0.6.12
DGraph-Fin Dateset
The dataset DGraph-Fin should be download and placed under the folder ./dataset/DGraph/raw.
Run Command
python main.py --model transformer --epochs 1000 --runs 10
Results
Performance on DGraph-Fin (10 runs):
Best Valid AUC: 0.7717 ± 0.0017
Best Test AUC: 0.7838 ± 0.0015
No Description
Python