Official repository for PEARL: A Review-driven Persona-Knowledge Grounded Conversational Recommendation Dataset accepted at ACL Findings 2024.
- Set environment
cd PEARL
conda create -n [YOUR_CONDA_ENV] python=3.10
conda activate [YOUR_CONDA_ENV]
pip install -r requirements.txt
- Create an OpenAI API key here.
- Download
resources.zipfrom gdrive. - Unzip
resources.zip. - Extract
vector_store_per_movie.tar.gzfile. mkdir -p data- Place
vector_store_per_movie,abstracts_per_user.json, anddialogue_input_data.jsonunderdata. - Modify
datagen/config/personal_info.json
- Replace
[YOUR_API_KEY_NAME]and[YOUR_OPENAI_API_KEY]with your OpenAI API credentials. - You may add more API keys to run the code more efficiently (make sure to set the API key name correctly in the script).
- Modify
scripts/generate_turn_by_turn.sh
- Replace
[YOUR_API_KEY_NAME]with one of the keys fromdatagen/config/personal_info.json.
- Run script
sh scripts/generate_turn_by_turn.sh