This repo contains two things:
- a pip-installable package that can be used to generate questions
- scripts that were used to generate the results of the ConFIRM paper
To install this package:
pip install git+https://github.com/WilliamGazeley/ConFIRM.git
For some examples of how this package can be used, it's recommended to take a look at the `tests/`` folder
In order to keep the package light-weight and usable for general cases, we separate the features/formatting that is specific to the ConFIRM paper.
Usage:
# Make sure the ConFIRM package is installed
pip install -r scripts/requirements.txt
bash scripts/run_question_generation.sh # Add your LLM provider credentials
# NOTE: download and place the Personage dataset under /datasets/personage-nlg
bash scripts/run_rephrasing.sh # Update with your filenames
# Finetuning
python wandb_scripts/train/lora_train.py
To run your own experiments, just use the package instead.