Klue Bert Base Re

🚀 KLUE関係抽出タスク用bert-base

KLUE REデータセットを使用してklue/bert-baseをファインチューニングしました。

• KLUEベンチマーク公式Webページ
• KLUE公式Github
• KLUE RE Github
• 無料GPUでKLUE REを実行する: Ainizeワークスペース

🚀 クイックスタート

このモデルは、KLUE関係抽出タスクに特化したbert-baseモデルです。以下のリンクを通じて関連情報や無料GPUでの実行環境を利用できます。

💻 使用例

基本的な使用法

from transformers import AutoTokenizer, AutoModelForSequenceClassification

tokenizer = AutoTokenizer.from_pretrained("ainize/klue-bert-base-re")
model = AutoModelForSequenceClassification.from_pretrained("ainize/klue-bert-base-re")

# Add "<subj>", "</subj>" to both ends of the subject object and "<obj>", "</obj>" to both ends of the object object.
sentence = "<subj>손흥민</subj>은 <obj>대한민국</obj>에서 태어났다."

encodings = tokenizer(sentence, 
                      max_length=128, 
                      truncation=True, 
                      padding="max_length", 
                      return_tensors="pt")

outputs = model(**encodings)

logits = outputs['logits']

preds = torch.argmax(logits, dim=1)

📚 ドキュメント

私たちについて

• Teachable NLP - コードを書かずに独自のテキストでNLPモデルをトレーニングできます。
• Ainize - 無料GPUを使用してMLプロジェクトをデプロイできます。