Summarizer Cnndm

🚀 要約モデル

cnn - dailymailデータセットの10000サンプルを使用して要約タスクに対して微調整されたBartForConditionalGenerationモデルです。

🚀 クイックスタート

このモデルは要約タスクに特化しており、cnn - dailymailデータセットを用いて微調整されています。

✨ 主な機能

• BartForConditionalGenerationモデルを使用した要約機能
• PyTorchモデルとして利用可能

📦 インストール

このモデルはtransformersライブラリを利用して使用できます。transformersライブラリがインストールされている必要があります。

💻 使用例

基本的な使用法

from transformers import AutoTokenizer, AutoModelWithLMHead, pipeline

tokenizer = AutoTokenizer.from_pretrained("yuvraj/summarizer-cnndm") 
AutoModelWithLMHead.from_pretrained("yuvraj/summarizer-cnndm")

summarizer = pipeline('summarization', model=model, tokenizer=tokenizer)
summarizer("<Text to be summarized>")

🔧 技術詳細

このモデルはBartForConditionalGenerationモデルを使用しており、cnn - dailymailデータセットの10000サンプルを用いて要約タスクに対して微調整されています。

📄 ライセンス

元のREADMEにライセンス情報は記載されていません。

⚠️ 重要提示

このモデルは小さなデータセットで学習されているため、性能に制限がある可能性があります。