Hugging Face's logo

recobo
/
chemical-bert-uncased-pharmaceutical-chemical-classifier

Text Classification
Transformers
PyTorch
Safetensors
English
bert
buy-intent
sell-intent
consumer-intent
Inference Endpoints
Model card Files Community
1
chemical-bert-uncased-pharmaceutical-chemical-classifier / README.md
shahrukhx01's picture
shahrukhx01
Update README.md
3d43a5f
preview code
|
raw history blame
1.12 kB
metadata
language: en
tags:
  - buy-intent
  - sell-intent
  - consumer-intent
widget:
  - text: >-
      Coronavirus disease (COVID-19) is an infectious disease caused by the
      SARS-CoV-2 virus.

Chemical domain language model finetuned on 13K Chemical, and 14K Pharma Wikipedia articles broken into paragraphs.

Chemical vs Pharmaceutical Domain Document Classifier

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 => "CHEMICAL"
LABEL_1 => "PHARMACEUTICAL"

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")