Keras layers for machine learning on graph structures
Project description
Neural fingerprint (nfp)
Keras layers for end-to-end learning on molecular structure. Based on Keras, Tensorflow, and RDKit. Source code used in the study Message-passing neural networks for high-throughput polymer screening
Related Work
- Convolutional Networks on Graphs for Learning Molecular Fingerprints
- Neural Message Passing for Quantum Chemistry
- Relational inductive biases, deep learning, and graph networks
- Neural Message Passing with Edge Updates for Predicting Properties of Molecules and Materials
(Main) Requirements
- rdkit
- tensorflow
Getting started
This library extends Keras with additional layers for handling molecular structures (i.e., graph-based inputs). There a strong familiarity with Keras is recommended.
An overview of how to build a model is shown in examples/solubility_test_graph_output.ipynb
. Models can optionally
include 3D molecular geometry; a simple example of a network using 3D geometry is found
in examples/model_3d_coordinates.ipynb
.
The current state-of-the-art architecture on QM9 (published in [4]) is included in examples/schnet_edgeupdate.py
. This
script requires qm9 preprocessing to be run before the model is evaluated with examples/preprocess_qm9.py
.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
File details
Details for the file nfp-0.3.12.tar.gz
.
File metadata
- Download URL: nfp-0.3.12.tar.gz
- Upload date:
- Size: 32.1 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.0 CPython/3.8.12
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | c9f1c3b2fc0163b288fe2285a99f9a694cf898bb8a6ad9c89c88c8e295d4dc26 |
|
MD5 | c0962a4a031c68f1189670e3af629f00 |
|
BLAKE2b-256 | a6eecfbebd2150914940827eda5c0c6de620d2332415683b8a59db03ae535b2a |
File details
Details for the file nfp-0.3.12-py3-none-any.whl
.
File metadata
- Download URL: nfp-0.3.12-py3-none-any.whl
- Upload date:
- Size: 19.4 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.0 CPython/3.8.12
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | e78cbcd9e63e56df5c3548f512ab8db53aa08f06e65eb4224f74b40d804b7a6b |
|
MD5 | 05f5c8065f05381d29f9cea678a95dd1 |
|
BLAKE2b-256 | a396b40f3c4234dba1e3a2fad070df6eb94b3e3b38a197184d9f36dc9bedf0ae |