Downloads · 30 days
12
8% of all-time downloads
Yousuf7/ChemBERT-BBB-Permeability
ChemBERT-BBB-Permeability is a text classification model from Yousuf7. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as mit.
Fine-tuned ChemBERTa-77M-MLM on the BBBP dataset for binary classification of blood-brain barrier (BBB) permeability (permeable vs non-permeable).
Downloads · 30 days
12
8% of all-time downloads
All-time downloads
143
Public
Parameters
3.4M
13.7 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors13.7 MB · 100%
From the Hugging Face model README
Fine-tuned ChemBERTa-77M-MLM on the BBBP dataset for binary classification of blood-brain barrier (BBB) permeability (permeable vs non-permeable).
Training was done manually with PyTorch on a MacBook Air M1 (CPU only, due to accelerate/MPS compatibility issues).
Try the model right now without any code:
Gradio Space → BBB Predictor Demo
Features:
from transformers import AutoTokenizer, AutoModelForSequenceClassification
model_id = "Yousuf7/ChemBERT-BBB-Permeability"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForSequenceClassification.from_pretrained(model_id, num_labels=2)