Chemical Bert Uncased Pharmaceutical Chemical Classifier

🚀 化學與製藥領域文檔分類器

本項目是一個化學領域語言模型，在13000篇化學和14000篇製藥領域的維基百科文章（按段落拆分）上進行了微調，可用於對化學和製藥領域的文檔進行分類。

🚀 快速開始

本分類器是在化學領域語言模型基礎上，通過大量化學和製藥領域的維基百科文章段落數據進行微調得到的。以下是模型在訓練、驗證和測試集上的表現：

📦 數據集

帶有劃分的數據集可以在 https://www.kaggle.com/shahrukhkhan/pharma-vs-chemicals-domain-classification 找到。

📚 標籤映射

• LABEL_0 => "製藥"
• LABEL_1 => "化學"

💻 使用示例

基礎用法

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