Chemical Bert Uncased Pharmaceutical Chemical Classifier

🚀 化学と製薬ドメインの文書分類器

化学ドメインの言語モデルで、13,000件の化学関連と14,000件の製薬関連のウィキペディア記事を段落に分割して微調整したものです。

🚀 クイックスタート

このモデルは、化学と製薬のドメイン文書を分類するために設計されています。以下の表は、トレーニング損失、検証精度、テスト精度を示しています。

📦 インストール

このモデルを使用するには、transformersライブラリが必要です。以下のコマンドでインストールできます。

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

📚 ドキュメント

データセット

分割されたデータセットは、https://www.kaggle.com/shahrukhkhan/pharma-vs-chemicals-domain-classification で見つけることができます。

ラベルマッピング

LABEL_0 => "PHARMACEUTICAL"
LABEL_1 => "CHEMICAL"