Response Content Units: Medical Open-Response Question Answering
This project uses Pixi for dependency and task management.
Before running extraction/evaluation tasks, make sure both are set up:
- Dataset files under
data/rcu-en/(see Dataset) - API keys for LLM tasks (see Environment variables)
pixi installpixi run run-gpt
pixi run run-gemini
pixi run run-qwen
pixi run run-deepseek
pixi run run-azure-deepseek
pixi run run-rule
pixi run train-mrc
pixi run run-mrc # Requires a trained model
pixi run run-biobert # Similarity-based extraction using BioBERT embeddingspixi run python -m src.extraction.cli --help
pixi run python -m src.extraction.cli gpt --model-name gpt-4o
pixi run python -m src.extraction.cli mrc-train --max-steps 5pixi run eval-medical-directives-classification
pixi run eval-question-classification
pixi run eval-binary-answer-classification
pixi run eval-question-extraction
pixi run eval-answer-extraction
pixi run eval-medical-directives-extraction
pixi run eval-prognosis-extraction
pixi run similarity-biobert
pixi run similarity-tfidfpixi run python -m src.eval.cli --help
pixi run python -m src.eval.cli medical-directives-classification
pixi run python -m src.eval.cli run answer-extractionpixi run test
pixi run test-verboseThere is currently no automated dataset download script in this repository.
Dataset source:
- https://osf.io/kcv2n/files/osfstorage (under
rcu-annotations-gem2026directory)
Expected location:
data/rcu-en/
Expected file naming pattern:
<dataset>_<split>.json(for example:iiyi_test_gold.json,woundcare_valid_systems.json)
If these files are missing, extraction/evaluation commands will fail or skip missing prediction files.
LLM extraction tasks require API keys.
Required keys:
AZURE_API_KEY(GPT, Azure DeepSeek)QWEN_API_KEYDEEPSEEK_API_KEYGOOGLE_API_KEY
You can provide keys in either of these ways:
.envfile at project root (recommended)- Exported shell environment variables
Extraction/eval CLIs auto-load .env from the project root.
- Copy
.env.exampleto.env - Fill in your keys
cp .env.example .envexport AZURE_API_KEY=...
export QWEN_API_KEY=...
export DEEPSEEK_API_KEY=...
export GOOGLE_API_KEY=...Then run tasks normally with Pixi (for example, pixi run extract-gpt).
Note: pixi run --clean-env ... will not inherit shell variables unless you explicitly provide them.
The Pixi workspace targets Python >=3.10,<3.13 for compatibility with current ML dependencies.