The dependencies involved have been written down in requirements.txt
run as follows:
pip install -r requirements.txt
- cora
- cora.cites
- cora.content
The .py files
in data_loader package
are as follows:
- loader.py : loading cora dataset, extracting node features and edge information, and calculating D-1/2 A D-1/2...
The .py files
in basic_version package
are as follows:
- layer.py : Defined Graph Convolution Layer (GraphConv)
- model.py : Defined GCN Model which composed by GraphConv
- evaluator.py : Used to evaluate our model
The .py files
in pyg_version package
are as follows:
- model.py : Defined GCN Model composed by
GCNConv
which provided by torch_geometric - evaluator.py : Used to evaluate our model