Bart Summarisation

🚀 bart-large-cnn-samsum

bart-large-cnn-samsum 模型是使用亞馬遜 SageMaker 和全新的 Hugging Face 深度學習容器訓練而成。該模型可用於抽象文本摘要任務，為文本摘要提供了有效的解決方案。

🚀 快速開始

此模型使用亞馬遜 SageMaker 和新的 Hugging Face 深度學習容器進行訓練。更多信息請查看：

• 🤗 Transformers 文檔：亞馬遜 SageMaker
• 示例筆記本
• 亞馬遜 SageMaker 的 Hugging Face 文檔
• Python SDK SageMaker 的 Hugging Face 文檔
• 深度學習容器

✨ 主要特性

• 訓練平臺：使用 Amazon SageMaker 和 Hugging Face 深度學習容器進行訓練。
• 適用任務：適用於抽象文本摘要任務。

📦 安裝指南

文檔未提及具體安裝步驟，故跳過此章節。

💻 使用示例

基礎用法

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)

📚 詳細文檔

超參數

{
    "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
}

結果

📄 許可證

本模型採用 Apache-2.0 許可證。