Roberta Base MITmovie

🚀 roberta-base + 映画のNERタスク

このプロジェクトは、MIT Movieデータセットを用いてNERタスク向けに訓練されたRoberta Baseモデルを提供します。

🚀 クイックスタート

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

✨ 主な機能

• 映画関連のデータセットを用いたNERタスクの実行
• Roberta Baseモデルをベースにした高精度な命名エンティティ認識

📦 インストール

インストールに関する具体的な手順は提供されていません。

💻 使用例

基本的な使用法

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 Movieでの評価

• epoch = 5.0
• eval_accuracy = 0.9476
• eval_f1 = 0.8853
• eval_loss = 0.2208
• eval_mem_cpu_alloc_delta = 17MB
• eval_mem_cpu_peaked_delta = 2MB
• eval_mem_gpu_alloc_delta = 0MB
• eval_mem_gpu_peaked_delta = 38MB
• eval_precision = 0.8833
• eval_recall = 0.8874
• eval_runtime = 0:00:03.62
• eval_samples = 1955

GitHubリポジトリ

• Domain-Adaptation Project

📄 ライセンス

このプロジェクトは cc-by-4.0 ライセンスの下で提供されています。