Skip to content
Snippets Groups Projects
Unverified Commit a878adf5 authored by MilanShao's avatar MilanShao Committed by GitHub
Browse files

Update README.md

parent 44532cc6
No related branches found
No related tags found
No related merge requests found
......@@ -10,6 +10,12 @@ Dependencies:
pip install torch --index-url https://download.pytorch.org/whl/cu117
pip install -r requirements.txt
```
## Data
```
The datasets for Luxemburg and the Smart Bridge Simulated Dataset in this repository represent 5 percent of the original data.
In the case of the Luxemburg dataset the 5 percent have even been to much to load it up in github. Therefore this link heads to the dataset: [https://www.dropbox.com/scl/fo/dk13szw9kf9nmlk5o31p2/h?dl=0&rlkey=j3qx3atdq2r4028gzhgd0y59s](https://www.dropbox.com/scl/fo/dk13szw9kf9nmlk5o31p2/h?dl=0&rlkey=j3qx3atdq2r4028gzhgd0y59s)
Please store the Luxemburg dataset in the data folder.
```
## How to use
......@@ -90,4 +96,4 @@ python train.py --model {Model} --dataset simulated_smart_bridge
| ModeConvLaplace | 92.43 | 88.07 |
| ChebConv | 82.15 | 83.89 |
| AGCRN | 92.26 | 87.76 |
| MtGNN | 91.19 | 86.78 |
\ No newline at end of file
| MtGNN | 91.19 | 86.78 |
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment