Kf Deberta Base Cross Nli

🚀 零樣本分類模型 - Transformers庫

本項目基於Transformers庫構建，使用kor - nli和klue數據集進行訓練，可用於零樣本分類任務，在KLUE - NLI測試集上取得了不錯的準確率。

🚀 快速開始

本模型使用kor - nli和klue - nli數據集進行訓練。 測試準確率（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許可證。