Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
AnoushkaVyas authored Mar 15, 2022
1 parent ad0faf1 commit b0d797a
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -60,7 +60,7 @@ To train a Hyperbolic Graph Convolutional Networks model for node classification

```python
cd GraphZoo
python graphzoo/trainers/train.py --task nc --dataset cora --model HGCN --lr 0.01 --dim 16 --num-layers 2 --act relu --bias 1 --dropout 0.5 --weight-decay 0.001 --manifold PoincareBall --log-freq 5 --cuda 0 --c None
python graphzoo/trainers/train.py --task nc --dataset cora --datapath <your datapath> --model HGCN --lr 0.01 --dim 16 --num-layers 2 --act relu --bias 1 --dropout 0.5 --weight-decay 0.001 --manifold PoincareBall --log-freq 5 --cuda 0 --c None
```
## Customizing Input Arguments

Expand Down Expand Up @@ -246,4 +246,4 @@ Some of the code was forked from the following repositories.
## License
[MIT](https://opensource.org/licenses/MIT)

Copyright (c) 2022 Anoushka Vyas
Copyright (c) 2022 Anoushka Vyas

0 comments on commit b0d797a

Please sign in to comment.