Skip to content

Latest commit

 

History

History
23 lines (19 loc) · 450 Bytes

README.md

File metadata and controls

23 lines (19 loc) · 450 Bytes

Run model

Environment setup

Requires pytorch, pytorch geometric, matplotlib and tqdm.

Train model

python ./GCNModel/gcn_model.py DATASET_NAME

Evaluate model

In ./GCNModel/evaluate.ipynb notebook, change dataset_name to the correct name.