Code implementation for fake news detection using GAT, GCN and GraphSAGE
- Numpy
- Pytorch (
torch_geometric
) - scikit-learn (
sklearn.metrics
)
Download dataset using this link
python run.py
For changing models you can pass --model='sage' or set value manually. The allowed values for models are:
- sage
- gcn
- gat
Other available args are as follows:
- --batch_size
- --lr
- --hidden_size
- --epochs
- --feature