Skip to content

Graph neural network (GNN) for molecular property prediction (3D structure)

License

Notifications You must be signed in to change notification settings

masashitsubaki/molecularGNN_3Dstructure

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

43 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Graph neural network (GNN) for molecular property prediction (3D structure)

Important: this repository will not be further developed and maintained because we have shown and believe that graph neural networks or graph convolutional networks are incorrect and useless for modeling molecules (see our paper in NeurIPS 2020). Please consider switching to our new and simple machine learning model called quantum deep field.

This code is a simpler version (different from the original paper) of our GNN model and its implementation for "Fast and Accurate Molecular Property Prediction: Learning Atomic Interactions and Potentials with Neural Networks (The Journal of Physical Chemistry Letters, 2018)" in PyTorch.

We show an example of the learning curve, which uses a subset of the QM9 dataset (the molecular size is less than 14 atoms), as follows.

This result can be completely reproduced by our code and one command (see "Usage").

Characteristics of our implementation

  • This code is easy-to-use for beginners. The requirement is only PyTorch.
  • Preprocessing a dataset and learning a GNN model can be done by only one command, "bash train.sh."
  • If you prepare another dataset with the same format as seen in the directory, dataset/QM9/data.txt, you can learn a GNN model with your dataset.

Requirements

  • PyTorch (of course numpy and scipy)

Usage

We provide two major scripts in the main directory as follows.

  • "preprocessing.py" creates tensor data from original text data (see dataset/QM9/data.txt).
  • "train.py" trains a GNN model using the preprocessed data to predict a molecular property.

You can easy to train a GNN model by the following commands.

Clone our repository,

git clone https://github.com/masashitsubaki/molecularGNN_3Dstructure.git

change directory,

cd molecularGNN_3Dstructure/main

and run the bash file for training.

bash train.sh

An image of running on google colaboratory is as follows.

You can also change the model hyperparameters described in train.sh (e.g., the dimensionality, number of hidden layers, and batch size).

Try to learn various GNN models to find your own best model for your dataset!

Learning a GNN with your dataset

In the dataset directory, we provide a subset of the QM9 dataset (see dataset/QM9/data.txt), which the format is as follows.

If you prepare a dataset with the same format (any molecular property can be used!), you can learn a GNN model with your dataset.

How to cite

@article{tsubaki2018fast,
  title={Fast and Accurate Molecular Property Prediction: Learning Atomic Interactions and Potentials with Neural Networks},
  author={Tsubaki, Masashi and Mizoguchi, Teruyasu},
  journal={The journal of physical chemistry letters},
  volume={9},
  number={19},
  pages={5733--5741},
  year={2018},
  publisher={ACS Publications}
}

About

Graph neural network (GNN) for molecular property prediction (3D structure)

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published