Skip to main content

Keras layers for machine learning on graph structures

Project description

Build Status PyPI version DOI

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

  1. Convolutional Networks on Graphs for Learning Molecular Fingerprints
  2. Neural Message Passing for Quantum Chemistry
  3. Relational inductive biases, deep learning, and graph networks
  4. Neural Message Passing with Edge Updates for Predicting Properties of Molecules and Materials

(Main) Requirements

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


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

nfp-0.3.12.tar.gz (32.1 kB view details)

Uploaded Source

Built Distribution

nfp-0.3.12-py3-none-any.whl (19.4 kB view details)

Uploaded Python 3

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

Hashes for nfp-0.3.12.tar.gz
Algorithm Hash digest
SHA256 c9f1c3b2fc0163b288fe2285a99f9a694cf898bb8a6ad9c89c88c8e295d4dc26
MD5 c0962a4a031c68f1189670e3af629f00
BLAKE2b-256 a6eecfbebd2150914940827eda5c0c6de620d2332415683b8a59db03ae535b2a

See more details on using hashes here.

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

Hashes for nfp-0.3.12-py3-none-any.whl
Algorithm Hash digest
SHA256 e78cbcd9e63e56df5c3548f512ab8db53aa08f06e65eb4224f74b40d804b7a6b
MD5 05f5c8065f05381d29f9cea678a95dd1
BLAKE2b-256 a396b40f3c4234dba1e3a2fad070df6eb94b3e3b38a197184d9f36dc9bedf0ae

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page