This is a PyTorch implementation.
Install pytorch and torchvision.
Four benchmark datasets (Tox21, HIV, Freesolv and Lipophilicity) are utilized in this study to evaluate the predictive performance of built graph convolutional networks. They are all downloaded from the MoleculeNet that hold various benchmark datasets for molecular machine learning.
Datasets are also provided in folder "Data".
Open the folder "eagcn_pytorch".
When you train the model for classification tasks, you can use:
python train.py
When you train the model for regression tasks, you can use:
python train_reg.py
The process to build the weighted adjacency tensor is:
Edge attention convolution is:
Our experiments evaluate the property prediction on standard supervised classification and regression tasks. We design our experiments with the goals of 1) verifying the improvement of our method compared with baseline methods, and 2) comparing two different architectures of our method. We adapt two edge attention layers and three fully connected layers for graph classification and regression.
Code is inspired by GCN and conv_qsar_fast