muti-Question Generation model

Overview

This repo is the official implementation of Diversity Enhanced Narrative Question Generation for Storybooks.

Tested Environment

• Python 3.10.6
• Pytorch-lightning 1.7.7

Instructions

First, install the Python dependencies:

pip install -r requirements.txt

Second, Download the checkpoint from the following Google Drive link:

mQG

https://drive.google.com/file/d/1-GQzRTfiUASz-de6Ikvz10xnNNO1MOYY/view?usp=share_link

Answerability Evaluation Model

https://drive.google.com/file/d/1-HCf8gJZ7V3uPgb3dqDDNcmnOi0MeS7U/view?usp=share_link

Third, run mQG_generate.py

If you'd like to train the model, you can download the FairytaleQA dataset from here.