Distilbert Base Uncased Squad2 With Ner Mit Restaurant With Neg With Repeat

🚀 distilbert-base-uncased-squad2-with-ner-mit-restaurant-with-neg-with-repeat

本模型是在squad_v2和mit_restaurant數據集上對 twmkn9/distilbert-base-uncased-squad2 進行微調後的版本，可用於特定的令牌分類任務。

🚀 快速開始

本模型可直接用於相關的令牌分類任務，你可以使用以下代碼加載模型：

from transformers import AutoModelForTokenClassification, AutoTokenizer

model_name = "distilbert-base-uncased-squad2-with-ner-mit-restaurant-with-neg-with-repeat"
model = AutoModelForTokenClassification.from_pretrained(model_name)
tokenizer = AutoTokenizer.from_pretrained(model_name)

✨ 主要特性

• 基於預訓練模型twmkn9/distilbert-base-uncased-squad2進行微調，在squad_v2和mit_restaurant數據集上進行訓練。
• 可用於令牌分類任務。

📦 安裝指南

使用前，請確保你已經安裝了以下依賴庫：

pip install transformers==4.8.2 torch==1.8.1+cu111 datasets==1.8.0 tokenizers==0.10.3

🔧 技術細節

訓練超參數

訓練過程中使用了以下超參數：

• 學習率（learning_rate）：2e-05
• 訓練批次大小（train_batch_size）：16
• 評估批次大小（eval_batch_size）：16
• 隨機種子（seed）：42
• 優化器（optimizer）：Adam，其中betas=(0.9, 0.999)，epsilon=1e-08
• 學習率調度器類型（lr_scheduler_type）：線性
• 訓練輪數（num_epochs）：5

框架版本

• Transformers：4.8.2
• Pytorch：1.8.1+cu111
• Datasets：1.8.0
• Tokenizers：0.10.3

📄 信息表格