Skip to content
/ EAGCN Public

Multi-View Spectral Graph Convolution with Consistent Edge Attention for Molecular Modeling

Notifications You must be signed in to change notification settings

Luckick/EAGCN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

49 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

EAGCN

This is a PyTorch implementation of paper "Multi-View Spectral Graph Convolution with Consistent Edge Attention for Molecular Modeling" published at Neurocomputing.

Installation

Install pytorch and torchvision.

Train EAGCN model

Dataset

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".

Train the model

Open the folder "eagcn_pytorch".

When you train the model, you can use:

python train.py

support files:
EAGCN_dataset.py: pre-processing data
neural_fp.py: from smiles to graph
layers.py: define layers
models.py: define models
utils.py: other tools

Visualization Tools

check_model.py: check parameters (edge attention for each layer).
mol_to_vec.py: visualize the molecule in 2D space, compare with other molecules which have similiar SMILEs.
plot.py: show model training process.
tsnes.py: tsne visualization about atom subtype, also provide umap option.
kmeans_atomrep.py: kmeans clustering for atom subtype.
plot_molecule.py: plot single molecule.

Citation

If you use this repository, e.g., the code and the datasets, in your research, please cite the following paper:

@article{shang2021multi,
  title={Multi-view spectral graph convolution with consistent edge attention for molecular modeling},
  author={Shang, Chao and Liu, Qinqing and Tong, Qianqian and Sun, Jiangwen and Song, Minghu and Bi, Jinbo},
  journal={Neurocomputing},
  volume={445},
  pages={12--25},
  year={2021},
  publisher={Elsevier}
}

Acknowledgments

Code is inspired by GCN and conv_qsar_fast

About

Multi-View Spectral Graph Convolution with Consistent Edge Attention for Molecular Modeling

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages