Skip to content
DimeNet model, as proposed in "Directional Message Passing for Molecular Graphs" (ICLR 2020)
Python Jupyter Notebook
Branch: master
Clone or download

README.md

Directional Message Passing Neural Network (DimeNet)

Reference implementation of the DimeNet model proposed in the paper:

Directional Message Passing for Molecular Graphs
by Johannes Klicpera, Janek Groß, Stephan Günnemann
Published at ICLR 2020.

Run the code

This repository contains a notebook for training the model (train.ipynb) and for generating predictions on the test set with a trained model (predict.ipynb). It also contains a script for training the model on a cluster with Sacred and SEML (train_seml.py). Note that this model is not optimized for runtime.

Architecture

Requirements

The repository uses these packages:

numpy
scipy
sympy>=1.5
tensorflow>=2.1
tensorflow_addons
tqdm

Contact

Please contact klicpera@in.tum.de if you have any questions.

Cite

Please cite our paper if you use the model or this code in your own work:

@inproceedings{klicpera_dimenet_2020,
  title = {Directional Message Passing for Molecular Graphs},
  author = {Klicpera, Johannes and Gro{\ss}, Janek and G{\"u}nnemann, Stephan},
  booktitle={International Conference on Learning Representations (ICLR)},
  year = {2020}
}
You can’t perform that action at this time.