Roberta Base MITmovie

🚀 roberta-base + 電影命名實體識別任務

本項目使用MIT電影數據集對RoBERTa Base模型進行訓練，以完成命名實體識別（NER）任務。該模型能夠精準識別電影相關文本中的命名實體，為電影數據處理和分析提供有力支持。

🚀 快速開始

以下是使用該模型進行命名實體識別的示例代碼：

model_name = "thatdramebaazguy/roberta-base-MITmovie"
pipeline(model=model_name, tokenizer=model_name, revision="v1.0", task="ner")

✨ 主要特性

• 模型架構：採用roberta-base語言模型。
• 適用語言：英文。
• 下游任務：命名實體識別（NER）。
• 訓練數據：MIT電影數據集。
• 評估數據：MIT電影數據集。
• 基礎設施：使用2塊Tesla v100進行訓練。
• 代碼示例：可參考 示例。

📦 安裝指南

文檔未提供具體安裝步驟，暫不展示。

💻 使用示例

基礎用法

model_name = "thatdramebaazguy/roberta-base-MITmovie"
pipeline(model=model_name, tokenizer=model_name, revision="v1.0", task="ner")

📚 詳細文檔

超參數設置

Num examples = 6253  
Num Epochs = 5 
Instantaneous batch size per device = 64
Total train batch size (w. parallel, distributed & accumulation) = 128  

性能評估

在MIT電影數據集上的評估結果

項目倉庫

• Domain-Adaptation項目

📄 許可證

本項目採用CC BY 4.0許可證。