yc-bench/run.sh
2026-03-19 17:52:49 -07:00

58 lines
2.4 KiB
Bash
Executable file

#!/usr/bin/env bash
set -euo pipefail
MODEL="gemini/gemini-3-flash-preview"
SEED=1
CONFIG="medium"
SLUG="${MODEL//\//_}"
# --- 1. Greedy bot baseline ---
uv run python scripts/bot_runner.py --bot greedy --config "$CONFIG" --seed "$SEED"
# --- 2. LLM run (delete stale DB first) ---
rm -f "db/${CONFIG}_${SEED}_${SLUG}.db"
uv run yc-bench run --model "$MODEL" --seed "$SEED" --config "$CONFIG" --company-name BenchCo --start-date 2025-01-01 --no-live
# --- 3. Plots ---
uv run python scripts/plot_results.py "results/yc_bench_result_${CONFIG}_${SEED}_${SLUG}.json" --plot funds
uv run python scripts/plot_results.py "results/yc_bench_result_${CONFIG}_${SEED}_${SLUG}.json" --plot prestige
uv run python scripts/plot_results.py "results/yc_bench_result_${CONFIG}_${SEED}_${SLUG}.json" --plot trust
# --- 4. Comparison plots (LLM vs greedy) ---
uv run python scripts/plot_results.py \
"results/yc_bench_result_${CONFIG}_${SEED}_${SLUG}.json" \
"results/yc_bench_result_${CONFIG}_${SEED}_greedy_bot.json" \
--plot funds --labels "LLM ($MODEL)" "greedy bot" \
--out plots/comparison_funds.png
uv run python scripts/plot_results.py \
"results/yc_bench_result_${CONFIG}_${SEED}_${SLUG}.json" \
"results/yc_bench_result_${CONFIG}_${SEED}_greedy_bot.json" \
--plot trust --labels "LLM ($MODEL)" "greedy bot" \
--out plots/comparison_trust.png
uv run python scripts/plot_results.py \
"results/yc_bench_result_${CONFIG}_${SEED}_${SLUG}.json" \
"results/yc_bench_result_${CONFIG}_${SEED}_greedy_bot.json" \
--plot prestige --labels "LLM ($MODEL)" "greedy bot" \
--out plots/comparison_prestige.png
# ============================================================================
# Quick reference commands (uncomment to use)
# ============================================================================
# --- Quick test run (50 turns max) ---
# rm -f db/fast_test_1_gemini_gemini-3-flash-preview.db
# uv run yc-bench run --model gemini/gemini-3-flash-preview --seed 1 --config fast_test --company-name BenchCo --start-date 2025-01-01
# --- Streamlit live dashboard (run alongside an LLM run) ---
# uv run streamlit run scripts/watch_dashboard.py -- "db/${CONFIG}_${SEED}_${SLUG}.db"
# --- Bot runner (all bots, all configs, all seeds) ---
# uv run python scripts/bot_runner.py
# --- Bot runner (single) ---
# uv run python scripts/bot_runner.py --bot greedy --config medium --seed 1
# --- Nuke all stale DBs (run after schema changes) ---
# rm -f db/*.db