Function write_svg

experiments/human_eval/draw_model_eval_ggbench_labeled.py:214–258

(rows: list[dict[str, object]])

def write_svg(rows: list[dict[str, object]]) -> None:
    xs = [float(row["auto"]) for row in rows]
    ys = [float(row["human"]) for row in rows]
    slope, intercept = regression(xs, ys)
    r = pearson(xs, ys)
    x1, x2 = min(xs), max(xs)
    y1, y2 = slope * x1 + intercept, slope * x2 + intercept

    def rgb(color: tuple[int, int, int]) -> str:
        return f"rgb({color[0]},{color[1]},{color[2]})"

    parts = [
        f'<svg xmlns="http://www.w3.org/2000/svg" width="{W}" height="{H}" viewBox="0 0 {W} {H}">',
        '<rect width="100%" height="100%" fill="white"/>',
        f'<line x1="{PLOT_L}" y1="{PLOT_T}" x2="{PLOT_L}" y2="{PLOT_B}" stroke="{rgb(AXIS)}" stroke-width="3"/>',
        f'<line x1="{PLOT_L}" y1="{PLOT_B}" x2="{PLOT_R}" y2="{PLOT_B}" stroke="{rgb(AXIS)}" stroke-width="3"/>',
    ]
    for tick in [65, 70, 75, 80, 85, 90, 95, 100]:
        y = y_pos(float(tick))
        parts.append(f'<line x1="{PLOT_L-10}" y1="{y}" x2="{PLOT_L}" y2="{y}" stroke="{rgb(AXIS)}" stroke-width="3"/>')
        parts.append(f'<text x="{PLOT_L-18}" y="{y+12}" font-family="Times New Roman" font-size="36" font-weight="700" text-anchor="end" fill="{rgb(TEXT)}">{tick}</text>')
    for tick in [63.5, 64.0, 64.5, 65.0]:
        x = x_pos(tick)
        parts.append(f'<line x1="{x}" y1="{PLOT_B}" x2="{x}" y2="{PLOT_B+13}" stroke="{rgb(AXIS)}" stroke-width="3"/>')
        parts.append(f'<text x="{x}" y="{PLOT_B+60}" font-family="Times New Roman" font-size="36" font-weight="700" text-anchor="middle" fill="{rgb(TEXT)}">{tick:.1f}</text>')
    parts.append(f'<text x="{PLOT_L-48}" y="{PLOT_T-52}" font-family="Times New Roman" font-size="42" font-weight="700" fill="{rgb(TEXT)}">Human</text>')
    parts.append(f'<text x="{PLOT_R+45}" y="{PLOT_B+20}" font-family="Times New Roman" font-size="42" font-weight="700" fill="{rgb(TEXT)}">ModelAuto</text>')
    parts.append(f'<line x1="{x_pos(x1)}" y1="{y_pos(y1)}" x2="{x_pos(x2)}" y2="{y_pos(y2)}" stroke="{rgb(RED)}" stroke-width="6"/>')
    for x, y in [(x1, y1), (x2, y2)]:
        parts.append(f'<circle cx="{x_pos(x)}" cy="{y_pos(y)}" r="10" fill="white" stroke="{rgb(BLUE)}" stroke-width="5"/>')
    for row in rows:
        key = str(row["key"])
        color = MODEL_COLORS[key]
        parts.append(f'<circle cx="{x_pos(float(row["auto"]))}" cy="{y_pos(float(row["human"]))}" r="11" fill="{rgb(color)}"/>')
    for row in rows:
        key = str(row["key"])
        color = MODEL_COLORS[key]
        dx, dy = LABEL_OFFSETS[key]
        parts.append(
            f'<text x="{x_pos(float(row["auto"]))+dx}" y="{y_pos(float(row["human"]))+dy+28}" '
            f'font-family="Times New Roman" font-size="31" font-weight="700" fill="{rgb(color)}">{row["name"]}</text>'
        )
    parts.append(f'<text x="{PLOT_R-430}" y="{PLOT_T+73}" font-family="Times New Roman" font-size="44" font-weight="700" fill="{rgb(RED)}">Pearson&apos;s r = {r:.4f}</text>')
    parts.append("</svg>")
    SVG_OUT.write_text("\n".join(parts), encoding="utf-8")


def write_csv(rows: list[dict[str, object]]) -> None: