-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathrun_sc_selection.sh
More file actions
executable file
·34 lines (31 loc) · 1.1 KB
/
run_sc_selection.sh
File metadata and controls
executable file
·34 lines (31 loc) · 1.1 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
#!/bin/bash
# --- Configuration ---
# Use the output of your generation baseline as PRE_PATH
if [ -z "${PYTHON_BIN:-}" ]; then
if [ -x ".venv/bin/python" ]; then
PYTHON_BIN=".venv/bin/python"
else
PYTHON_BIN="python3"
fi
fi
ARTIFACT_ROOT=${ARTIFACT_ROOT:-"artifacts"}
DATASET_TYPE=${DATASET_TYPE:-"bird"}
DATA_PATH=${DATA_PATH:-"data/bird/dev/mini_dev.json"}
DB_ROOT_PATH=${DB_ROOT_PATH:-"data/bird/dev/dev_databases"}
PRED_PATH=${PRED_PATH:-"${ARTIFACT_ROOT}/candidates/GPT-5_BIRD_Mini_Dev.json"}
OUTPUT_PATH=${OUTPUT_PATH:-"${ARTIFACT_ROOT}/selected/GPT-5_BIRD_Mini_Dev_SC.json"}
NUM_WORKERS=${NUM_WORKERS:-8}
TIMEOUT=${TIMEOUT:-30}
echo "Starting Self-Consistency (SC) Selection..."
echo "Dataset: $DATASET_TYPE"
echo "Input Candidates: $PRED_PATH"
echo "Output Selected: $OUTPUT_PATH"
echo "Workers: $NUM_WORKERS"
"$PYTHON_BIN" baseline/run_sc_selection.py \
--pred_path "$PRED_PATH" \
--dataset_type "$DATASET_TYPE" \
--data_path "$DATA_PATH" \
--db_root_path "$DB_ROOT_PATH" \
--output_path "$OUTPUT_PATH" \
--num_workers "$NUM_WORKERS" \
--timeout "$TIMEOUT"