模型:

thanathorn/mt5-cpe-kmutt-thai-sentence-sum

英文

mt5-cpe-kmutt-thai-sentence-sum

该代码库包含了用于泰语句子摘要的精调mT5-base模型。该模型的架构基于mT5模型,并在泰语文本摘要对上进行了精调。此外,这个项目是曼谷工业大学计算机工程学生的毕业项目。

在SimpleTransformer上的使用(已在版本0.63.4上测试)

from simpletransformers.t5 import T5Model, T5Args
from torch import cuda

model = T5Model("t5", "thanathorn/mt5-cpe-kmutt-thai-sentence-sum", use_cuda=cuda.is_available())

sentence = "simplify: ถ้าพูดถึงขนมหวานในตำนานที่ชื่นใจที่สุดแล้วละก็ต้องไม่พ้น น้ำแข็งใส แน่เพราะว่าเป็นอะไรที่ชื่นใจสุด"
prediction = model.predict([sentence])
print(prediction[0])

(请参考 Google Colab 上的示例)

得分

  • ROUGE-1: 61.7805
  • ROUGE-2: 45.9689
  • ROUGE-L: 59.3542

预期用途和限制

  • 您可以使用该模型进行泰语句子文本摘要。
  • 不适用于段落文本。