许可证: apache-2.0
标签:
- 摘要生成
- 训练生成
数据集:
- samsum
评估指标:
- rouge
基础模型: facebook/bart-base
模型索引:
- 名称: bart-base-finetuned-samsum-en
结果:
- 任务:
类型: 文本到文本生成
名称: 序列到序列语言建模
数据集:
名称: samsum
类型: samsum
参数: samsum
评估指标:
- 类型: rouge
值: 46.8825
名称: Rouge1
- 任务:
类型: 摘要生成
名称: 摘要生成
数据集:
名称: samsum
类型: samsum
配置: samsum
拆分: 测试集
评估指标:
- 类型: rouge
值: 45.0692
名称: ROUGE-1
已验证: true
验证令牌: eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJoYXNoIjoiYmJmZGIzMjdlODZhOThjMTFhYzM3NWJjOTNjMjkyN2U4NmE2YjViZmM0ZTBkNTU0NTc0YmZlMGY4NDI1NmM0YyIsInZlcnNpb24iOjF9.4p6xyMhFMia_Ms0bohyUchjrHhYvz7tH_O9LdeEK5FYF6pWEWs3Aw3i6LCc2HPnsge2ZKG-pzsRdbMJwO5w4Bg
- 类型: rouge
值: 20.9049
名称: ROUGE-2
已验证: true
验证令牌: eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJoYXNoIjoiNTY3OTVmM2E5ZmRjYzYzZmE3MGQwODNmMTk0MGJjMTRkMDQ2NGEwNGY0ZTA3OWJjODIzOWVkYTgwZjRjNzhkYiIsInZlcnNpb24iOjF9.YKFCR8fT38vuj9TA9wZHUNX6hmV0_m4yErKuuVvEBtWSwOmrzIrdSHVk40wSQomojTFFA9HRAZ_Ssf0rke_SAw
- 类型: rouge
值: 37.3128
名称: ROUGE-L
已验证: true
验证令牌: eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJoYXNoIjoiZDZhN2I5NDdjNTEwYzg2YTZlMWJkZDFmYmE0NDkzYmI4Mjg2YjVkYTQwMmEyMTEzZmI4YzBmMDg3OWY5MDQ0MiIsInZlcnNpb24iOjF9.RFrkndxIOt61Vs59zCdI09IXh349inPilhD-bXvpqfUKa92PTUNBqsumL8lzS4N7uEcfca1QYG4STEy_Efc9Dw
- 类型: rouge
值: 40.662
名称: ROUGE-LSUM
已验证: true
验证令牌: eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJoYXNoIjoiMGIxMjlkZjY3ZDNiOGQwM2FlNTExYjI5OGIwOGUxYTU4ZGFiNGRjOTU5MTFjMDQ2YWU2MzU3M2E0NTQ3Yjk4MiIsInZlcnNpb24iOjF9.AEJgh8jnUs0nyIsUj6mNxyqIzfFRVwqEowIMnK-xnSRbuMkCMEREeao4FSK2a0Cx5rNkXuEW4fBIRDrhyDKIBA
- 类型: 损失
值: 5.763935565948486
名称: 损失
已验证: true
验证令牌: eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJoYXNoIjoiYzc2Yzc5YjRjOGFjZmIwYTFiYzNiN2VmNzJiNjdlZmRhNjZlNzU4ZDc1OWE2MTk1MjQ5MGM0NDNmMDY3NTZmYiIsInZlcnNpb24iOjF9.r-Z0kcXf81IlDxT0yYoLjya7Uc3q6sJT0LxzhIdKVYc3mOoxPLMX1_vLQPX8fsIHIqPHPtKczfZ_z6mB0yptBg
- 类型: 生成长度
值: 18.4921
名称: 生成长度
已验证: true
验证令牌: eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJoYXNoIjoiY2NkYTg0MGFiYmJiZWI1ZjYzMGEzOGM1MGM2NjgyNzE2MTk2ZTc3MTQ2Mzg4Y2U5NTM3MDBiNTQ5YzUwNzAzYyIsInZlcnNpb24iOjF9.fn2kkheKjJJio6pcI0X0OWMEE08CuO35VrebfTNqTGJDV3U5O2f6Vf6RMo0d5b4N5X-E5-Ju-7k2CDxNi8CAAw
bart-base-finetuned-samsum-en
该模型是基于facebook/bart-base在samsum数据集上微调的版本。
在评估集上取得了以下结果:
- 损失: 2.3676
- Rouge1: 46.8825
- Rouge2: 22.0923
- Rougel: 39.7249
- Rougelsum: 42.9187
模型描述
需补充更多信息
预期用途与限制
需补充更多信息
训练与评估数据
需补充更多信息
训练过程
训练超参数
训练过程中使用了以下超参数:
- 学习率: 5.6e-05
- 训练批次大小: 10
- 评估批次大小: 10
- 随机种子: 42
- 优化器: Adam,参数为betas=(0.9,0.999),epsilon=1e-08
- 学习率调度器类型: 线性
- 训练轮数: 3
训练结果
训练损失 |
轮次 |
步数 |
验证损失 |
Rouge1 |
Rouge2 |
Rougel |
Rougelsum |
0.5172 |
1.0 |
300 |
2.1613 |
47.4152 |
22.8106 |
39.93 |
43.3639 |
0.3627 |
2.0 |
600 |
2.2771 |
47.2676 |
22.6325 |
40.1345 |
43.19 |
0.2466 |
3.0 |
900 |
2.3676 |
46.8825 |
22.0923 |
39.7249 |
42.9187 |
框架版本
- Transformers 4.19.2
- Pytorch 1.11.0+cu113
- Datasets 2.2.2
- Tokenizers 0.12.1