Kf Deberta Base Cross Nli

🚀 transformers

このライブラリはkor - nliとklueデータセットを用いて学習されたゼロショット分類モデルを提供します。高い精度でのテスト結果が得られています。

🚀 クイックスタート

このモデルはkor - nliとklueデータセットで学習されています。テスト精度(KLUE - NLI)は0.889667です。

💻 使用例

基本的な使用法

model_name = 'deliciouscat/kf-deberta-base-cross-nli'
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=3).to(device)

def encode(examples):
    return tokenizer(examples['premise'], examples['hypothesis'], truncation=True, padding='max_length', max_length=128)

📄 ライセンス

このプロジェクトはMITライセンスの下で公開されています。