context, a list of choices, and the answer_index of the one correct choice. The same shape covers two different tasks, and the models read the context differently for each.
Shape
- Completion stem
- Question
- Generic with style
The context is a sentence stem the correct choice continues, as in HellaSwag. The stem ends mid-sentence by design, so it is never judged for completeness.
answer_index outside the choices, or an unknown style.
Extra columns
Anything beyond the core fields goes inextra and comes back untouched:
What the models return
sdvm-audit-1
What it returns for this sample.
sdvm-fix-1
What it changes, and what it never does.
Examples
hellaswag.py for stems and mmlu.py for questions, end to end.