๐Ÿ“Š Predictive Benchmark Runner

February 8, 2026 ยท View on GitHub

This directory contains the LLM-based evaluation runner. It consumes ground-truth group files (generated by the data-prep pipeline), prompts an LLM to rank solutions, and produces per-task and cross-task grading reports.


๐Ÿง Functionality

  • ๐Ÿ“ฅ Loads ground-truth comparison groups (e.g., groups_<task>_n2.json).
  • ๐Ÿ“ Constructs Prompts using task descriptions and data analysis reports.
  • ๐Ÿง  Queries LLMs to predict the better solution (pairwise) or rank multiple solutions.
  • ๐Ÿ“ˆ Computes Metrics: Accuracy (for n=2), Spearman Correlation & Precision@k (for n>2).
  • ๐Ÿ’พ Saves Reports including full interaction logs and machine-readable JSONs.

Entrypoint: run_bench.py


๐Ÿƒ Quick Start

python -m skip_bench.grade.run_bench \
  --task-file /path/to/task_name.txt \
  --solutions-dir /path/to/solutions_root \
  --n 2 \
  --model deepseek-v3.2-thinking \
  --temperature 1.0 \
  --parallel 64 \
  --task-parallel 8 \
  --prompt-boost \
  --desc-dir /path/to/description \
  --da-dir /path/to/data_analysis/result \
  --cot

๐Ÿ“‚ Expected Inputs

  1. ๐Ÿ“‹ Task List (--task-file)

    • A text file with one task name per line.
  2. ๐Ÿ—‚๏ธ Solutions Root (--solutions-dir)

    • Structure: <solutions_dir>/<task>/ground_truth/groups_<task>_n*.json
  3. ๐Ÿ“š Task Resources (Optional but Recommended)

    • Mode 1 (Root): --tasks-root containing description/ and data_analysis/.
    • Mode 2 (Explicit): --desc-dir for descriptions and --da-dir for analysis.
    • Mode 3 (Raw): --raw-data-sample for raw data snippets.
  4. ๐ŸŽฏ Ground Truth Override

    • --groundtruth-file: Point to a specific JSON file (supports {task} placeholder).

โš™๏ธ Key Arguments

CategoryFlagDescription
Input--task-filePath to task list.
--solutions-dirRoot for solutions & ground truth.
--nGroup size (e.g., 2 for pairwise). 0 for all.
Model--modelModel name (e.g., gpt-4).
--temperatureSampling temperature.
Execution--parallelConcurrency per task.
--task-parallelNumber of tasks to run in parallel.
Prompting--prompt-boostEnhances instruction emphasis.
--cotEnaables Chain-of-Thought reasoning.
Output--partial-jsonSave checkpoints (default: on).

๐Ÿ“ค Outputs

Artifacts are written to <solutions_dir>/<task>/report/:

  • ๐Ÿ“„ Human-Readable Report: grade_report_<task>_... .txt (Metrics + Logs)
  • ๐Ÿค– Alignment JSON: alignment_<task>_... .json (Structured predictions)
  • ๐Ÿ’พ Partial Checkpoint: grade_results_partial.json
  • ๐Ÿ“Š Global Report: <solutions_dir>/report/grade_report_alltasks_*.txt

Notes

  • Group files are created by the data-prep pipeline (see the README in prepare_bench_subset).
  • If you pass --groundtruth-file, you must set --n to a non-zero value.
  • When using relative paths in group_entry.paths, keep --solutions-dir set so the runner can resolve files correctly.
  • The command examples above may not list all tunable flags. For a complete reference, run each script with --help or inspect the corresponding Python file and its argparse comments.