英文

lmqg/mt5-small-frquad-qag 模型卡片

该模型是通过 lmqg lmqg/qag_frquad (数据集名称:default)上进行问题和答案对生成任务进行微调的 google/mt5-small 的版本。

概述

用法

from lmqg import TransformersQG

# initialize model
model = TransformersQG(language="fr", model="lmqg/mt5-small-frquad-qag")

# model prediction
question_answer_pairs = model.generate_qa("Créateur » (Maker), lui aussi au singulier, « le Suprême Berger » (The Great Shepherd) ; de l'autre, des réminiscences de la théologie de l'Antiquité : le tonnerre, voix de Jupiter, « Et souvent ta voix gronde en un tonnerre terrifiant », etc.")
  • 使用 transformers
from transformers import pipeline

pipe = pipeline("text2text-generation", "lmqg/mt5-small-frquad-qag")
output = pipe("Créateur » (Maker), lui aussi au singulier, « le Suprême Berger » (The Great Shepherd) ; de l'autre, des réminiscences de la théologie de l'Antiquité : le tonnerre, voix de Jupiter, « Et souvent ta voix gronde en un tonnerre terrifiant », etc.")

评估

Score Type Dataset
QAAlignedF1Score (BERTScore) 77.23 default 12313321
QAAlignedF1Score (MoverScore) 52.36 default 12313321
QAAlignedPrecision (BERTScore) 76.76 default 12313321
QAAlignedPrecision (MoverScore) 52.19 default 12313321
QAAlignedRecall (BERTScore) 77.74 default 12313321
QAAlignedRecall (MoverScore) 52.54 default 12313321

训练超参数

在微调过程中使用了以下超参数:

  • 数据集路径:lmqg/qag_frquad
  • 数据集名称:default
  • 输入类型:['paragraph']
  • 输出类型:['questions_answers']
  • 前缀类型:无
  • 模型:google/mt5-small
  • 最大长度:512
  • 输出最大长度:256
  • epoch 数:13
  • 批次大小:8
  • 学习率:0.001
  • fp16:False
  • 随机种子:1
  • 梯度积累步数:8
  • 标签平滑:0.0

完整的配置信息可以在 fine-tuning config file 找到。

引用

@inproceedings{ushio-etal-2022-generative,
    title = "{G}enerative {L}anguage {M}odels for {P}aragraph-{L}evel {Q}uestion {G}eneration",
    author = "Ushio, Asahi  and
        Alva-Manchego, Fernando  and
        Camacho-Collados, Jose",
    booktitle = "Proceedings of the 2022 Conference on Empirical Methods in Natural Language Processing",
    month = dec,
    year = "2022",
    address = "Abu Dhabi, U.A.E.",
    publisher = "Association for Computational Linguistics",
}