Downloads · 30 days
0
tdc/BBB_Martins-CNN
BBB_Martins-CNN is a machine learning model from tdc. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for tdc. The card lists the license as bsd-2-clause.
As a membrane separating circulating blood and brain extracellular fluid, the blood-brain barrier (BBB) is the protective layer that blocks most foreign drugs. Thus the ability of a drug to penetrate the barrier to de…
Downloads · 30 days
0
Access
Public
Updated Apr 13, 2024
Repo size
9.8 MB
Likes
0
Public
Click a slice to open those files.
.pt9.8 MB · 100%
From the Hugging Face model README
As a membrane separating circulating blood and brain extracellular fluid, the blood-brain barrier (BBB) is the protective layer that blocks most foreign drugs. Thus the ability of a drug to penetrate the barrier to deliver to the site of action forms a crucial challenge in developing drugs for the central nervous system.
Binary classification. Given a drug SMILES string, predict the activity of BBB.
Total: 1,975 drugs
Install the following packages
pip install PyTDC
pip install DeepPurpose
pip install git+https://github.com/bp-kelley/descriptastorus
pip install dgl torch torchvision
You can also reference the colab notebook here
Random split with 70% training, 10% validation, and 20% testing
To load the dataset in TDC, type
from tdc.single_pred import ADME
data = ADME(name = 'BBB_Martins')
CNN is applying Convolutional Neural Network on SMILES string fingerprint. Model is tuned with 100 runs using Ax platform. To load the pre-trained model, type
from tdc import tdc_hf_interface
tdc_hf = tdc_hf_interface("BBB_Martins-CNN")
# load deeppurpose model from this repo
dp_model = tdc_hf.load_deeppurpose('./data')
tdc_hf.predict_deeppurpose(dp_model, ['YOUR SMILES STRING'])