run_all_splits.sh

#!/bin/bash
# 批量运行 10 个 locomo split：import + evaluate
# 每个 split 都会清空数据库重新导入，然后评估

set -e  # 遇到错误立即退出

SPLITS=("conv-26" "conv-30" "conv-41" "conv-42" "conv-43" "conv-44" "conv-47" "conv-48" "conv-49" "conv-50")

echo "=========================================="
echo "Running all 10 locomo splits"
echo "=========================================="
echo ""

for CONV_ID in "${SPLITS[@]}"; do
    echo ""
    echo "=========================================="
    echo "Processing ${CONV_ID}"
    echo "=========================================="
    echo ""

    # Step 1: Import data (自动清空数据库)
    echo "[1/2] Importing ${CONV_ID}..."
    python import_data.py -c "${CONV_ID}" <<EOF
y
EOF

    if [ $? -ne 0 ]; then
        echo "[ERROR] Import failed for ${CONV_ID}, skipping..."
        continue
    fi

    # Step 2: Evaluate
    echo ""
    echo "[2/2] Evaluating ${CONV_ID}..."
    python evaluate.py gen -c "${CONV_ID}"

    if [ $? -ne 0 ]; then
        echo "[ERROR] Evaluation failed for ${CONV_ID}"
    fi

    echo ""
    echo "[Done] ${CONV_ID} completed"
    echo ""
done

echo ""
echo "=========================================="
echo "All splits completed!"
echo "=========================================="