Bart Large Cnn Samsum

🚀 bart-large-cnn-samsum

這個模型是 facebook/bart-large-cnn 在 samsum 數據集 上的微調版本。它主要用於對話摘要生成，能夠將對話內容進行有效總結，為用戶提供簡潔的信息。

🚀 快速開始

這個模型是 facebook/bart-large-cnn 在 samsum 數據集 上的微調版本。它在評估集上取得了以下結果：

• 損失率：0.755
• Rouge1：43.6283
• Rouge2：19.3096
• Rougel：41.2140
• Rougelsum：37.2590

✨ 主要特性

• 微調模型：基於 facebook/bart-large-cnn 進行微調，更適配 samsum 數據集。
• 評估指標良好：在 Rouge 指標上表現出色，如 Rouge-1 達到 43.6283。

💻 使用示例

基礎用法

from transformers import pipeline
summarizer = pipeline("summarization", model="AdamCodd/bart-large-cnn-samsum")

conversation = '''Emily: Hey Alex, have you heard about the new restaurant that opened downtown?
Alex: No, I haven't. What's it called?
Emily: It's called "Savory Bites." They say it has the best pasta in town.
Alex: That sounds delicious. When are you thinking of checking it out?
Emily: How about this Saturday? We can make it a dinner date.
Alex: Sounds like a plan, Emily. I'm looking forward to it.                                       
'''
result = summarizer(conversation)
print(result)

📚 詳細文檔

訓練和評估數據

更多信息待補充。

訓練過程

訓練超參數

訓練過程中使用了以下超參數：

• 學習率：3e-05
• 訓練批次大小：4
• 評估批次大小：4
• 隨機種子：1270
• 優化器：AdamW，β1=0.9，β2=0.999，ε=1e-08
• 學習率調度器類型：線性
• 學習率調度器熱身步數：150
• 訓練輪數：1

訓練結果

框架版本

• Transformers 4.34.0
• Pytorch lightning 2.0.9
• Tokenizers 0.14.0

📄 許可證

本項目採用 Apache-2.0 許可證。

如果你想支持作者，可以點擊 這裡。