Skip to content

Commit

Permalink
added airport data
Browse files Browse the repository at this point in the history
  • Loading branch information
ines-chami committed May 1, 2020
1 parent 220cad5 commit b9dab7a
Show file tree
Hide file tree
Showing 5 changed files with 67,704 additions and 5 deletions.
5 changes: 5 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -57,6 +57,7 @@ The ```data/``` folder contains source files for:
* Cora
* Pubmed
* Disease
* Airport

To run this code on new datasets, please add corresponding data processing and loading in ```load_data_nc``` and ```load_data_lp``` functions in ```utils/data_utils.py```.

Expand Down Expand Up @@ -168,6 +169,10 @@ We provide examples of training commands used to train HGCN and other graph embe

```python train.py --task lp --dataset disease_lp --model HGCN --lr 0.01 --dim 16 --num-layers 2 --num-layers 2 --act relu --bias 1 --dropout 0 --weight-decay 0 --manifold PoincareBall --normalize-feats 0 --log-freq 5```

* Airport (Test ROC-AUC=97.43):

```python train.py --task lp --dataset airport --model HGCN --lr 0.01 --dim 16 --num-layers 2 --act relu --bias 1 --dropout 0.0 --weight-decay 0 --manifold PoincareBall --log-freq 5 --cuda 0 --c None```

#### Node classification

* Cora and Pubmed:
Expand Down
Binary file added data/airport/airport.p
Binary file not shown.
Binary file added data/airport/airport_alldata.p
Binary file not shown.
Loading

0 comments on commit b9dab7a

Please sign in to comment.