File size: 2,065 Bytes
7334e4c 631729d 7334e4c 631729d 7334e4c 631729d 7334e4c 631729d 7334e4c 53b3ce4 7334e4c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
---
library_name: keras
---
## Model description
This repo contains the model and the notebook for implementing Message Passing Neural Network (MPNN) to predict a molecular property known as blood-brain barrier permeability (BBBP). [Message-passing neural network (MPNN) for molecular property prediction](https://keras.io/examples/graph/mpnn-molecular-graphs/).
Full credits go to [Alexander Kensert](https://github.com/akensert)
Reproduced by [Vu Minh Chien](https://www.linkedin.com/in/vumichien/)
Motivation: Until now, more traditional methods, such as random forests, support vector machines, etc., have been commonly used to predict molecular properties. In contrast to GNNs, these traditional approaches often operate on precomputed molecular features such as molecular weight, polarity, charge, number of carbon atoms, etc. Although these molecular features prove to be good predictors for various molecular properties, it is hypothesized that operating on these more "raw", and "low-level", features could prove even better. In this tutorial, MPNN was used to prove that GNN is a good method to predict molecular properties by using the undirected graph of molecules
## Training and evaluation data
Information about the dataset can be found in A Bayesian Approach to Silico Blood-Brain Barrier Penetration Modeling and MoleculeNet: A Benchmark for Molecular Machine Learning. The dataset will be downloaded from MoleculeNet.org.
The dataset contains 2,050 molecules. Each molecule comes with a name, label, and SMILES string.
## Training procedure
### Training hyperparameters
The following hyperparameters were used during training:
| name | learning_rate | decay | beta_1 | beta_2 | epsilon | amsgrad | training_precision |
|----|-------------|-----|------|------|-------|-------|------------------|
|Adam|0.0005000000237487257|0.0|0.8999999761581421|0.9990000128746033|1e-07|False|float32|
## Model Plot
### View Model Demo
<details>
<summary>View Model Plot</summary>
</details> |