Bart Summarisation

🚀 bart-large-cnn-samsum

このモデルは、Amazon SageMakerと新しいHugging Faceのディープラーニングコンテナを使用して学習されました。詳細については以下を参照してください。

🚀 クイックスタート

このモデルは、Amazon SageMakerと新しいHugging Faceのディープラーニングコンテナを用いて訓練されました。詳細情報については以下のリンクを参照してください。

• 🤗 Transformers Documentation: Amazon SageMaker
• Example Notebooks
• Amazon SageMaker documentation for Hugging Face
• Python SDK SageMaker documentation for Hugging Face
• Deep Learning Container

✨ 主な機能

このモデルは、抽象的な文章要約タスクに特化しており、SAMSumコーパスを使用して訓練されています。

📚 ドキュメント

ハイパーパラメータ

{
    "dataset_name": "samsum",
    "do_eval": true,
    "do_predict": true,
    "do_train": true,
    "fp16": true,
    "learning_rate": 5e-05,
    "model_name_or_path": "facebook/bart-large-cnn",
    "num_train_epochs": 3,
    "output_dir": "/opt/ml/model",
    "per_device_eval_batch_size": 4,
    "per_device_train_batch_size": 4,
    "predict_with_generate": true,
    "seed": 7
}

💻 使用例

基本的な使用法

from transformers import pipeline
summarizer = pipeline("summarization", model="slauw87/bart-large-cnn-samsum")
conversation = '''Sugi: I am tired of everything in my life. 
Tommy: What? How happy you life is! I do envy you.
Sugi: You don't know that I have been over-protected by my mother these years. I am really about to leave the family and spread my wings.
Tommy: Maybe you are right.                                           
'''
nlp(conversation)

結果

📄 ライセンス

このモデルはApache-2.0ライセンスの下で提供されています。