QAFD-RAG / data /multihop /README.md
tarzanagh's picture
Add code: src, benchmarks, data, requirements, run.sh, README
8e874f5 verified
|
Raw
History Blame Contribute Delete
680 Bytes

Multi-hop QA Data

Datasets: MuSiQue, HotpotQA, 2WikiMultiHopQA (included in this directory).

Graph type: passage-entity (default) or entity (--graph_type entity)

Pre-built KGs

Pre-built KGs use nvidia-nv-embed-v2 (requires GPU, 16GB+ VRAM, auto-downloaded):

huggingface-cli download qafd/kg --repo-type dataset --include "multihop/*" --local-dir ./kg

Usage

# Uses downloaded KG, or builds from scratch if not found
python benchmarks/run.py --task multihop --dataset musique --questions 100

# Rebuild with openai-small (no GPU needed)
python benchmarks/run.py --task multihop --dataset musique --force_build --embedding openai-small