Downloads · 30 days
10
0% of all-time downloads
chemical-bert-uncased-pharmaceutical-chemical-classifier
recobo/chemical-bert-uncased-pharmaceutical-chemical-classifier
chemical-bert-uncased-pharmaceutical-chemical-classifier is a text classification model from recobo. Use it when you need a label for a piece of text. It is set up for transformers.
Chemical domain language model finetuned on 13K Chemical, and 14K Pharma Wikipedia articles broken into paragraphs.
All-time downloads
3K
Public
Parameters
110M
1.3 GB on disk
Likes
2
Public
Repo makeup
Click a slice to open those files.
.bin440 MB · 50%
Parameter types
How the weights are stored.
F32110M · 100%
Try a prompt
Model card
From the Hugging Face model README
Chemical vs Pharmaceutical Domain Document Classifier
Chemical domain language model finetuned on 13K Chemical, and 14K Pharma Wikipedia articles broken into paragraphs.
| Train Loss | Validation Acc. | Test Acc. |
|---|---|---|
| 0.17 | 0.928 | 0.927 |
Dataset
Dataset with splits can be found @ https://www.kaggle.com/shahrukhkhan/pharma-vs-chemicals-domain-classification
Label Mappings
LABEL_0 => "PHARMACEUTICAL" <br/> LABEL_1 => "CHEMICAL"
Usage in Transformers
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("recobo/chemical-bert-uncased-pharmaceutical-chemical-classifier")
model = AutoModelForSequenceClassification.from_pretrained("recobo/chemical-bert-uncased-pharmaceutical-chemical-classifier")
- Task
- Text Classification
- Library
- transformers
- Type
- bert
- Languages
- en
- Created
- Mar 2, 2022
- Updated
- Mar 23, 2023