feat(categorize): confidence calibration engine + measurement loop (cascade step 4) (#1784)
Turns the selector's raw confidence into a score that means what it says. - lib/agent/categorize/calibration.ts: the engine. Isotonic regression (pool-adjacent-violators, distribution-free + monotonic) over (confidence, was_correct) samples → a calibrator; plus reliabilityByBucket, ECE, and bandFor(). bandFor NEVER returns 'auto' without a fitted calibrator (no silent booking on an unproven score) and never auto-books above an amount cap. 12 engine tests (overconfidence pulled down, underconfidence lifted, monotonicity, ECE, band gating). - Measurement loop: migration categorize_calibration_samples (append-only, company-scoped RLS, confidence CHECK [0,1]) + POST /api/agent/categorize/ outcome logging one sample (proposed vs actually booked) fire-and-forget from QuickReviewDialog on a successful book (sandbox skipped). AiCategorizeProposal surfaces the proposal metadata via onProposal. - scripts/fit-categorize-calibration.ts (read-only): prints the reliability diagram + ECE + fitted calibrator once data has accumulated. Fitting needs a few hundred real outcomes, so nothing calibrates today — the loop starts collecting, and "säker" stays uncalibrated (no auto-book) until the data proves it. 131 unit tests green; RLS covered by a pg-real test. Co-authored-by: Jakob Wennberg <311770904+jakobwennberg-oss@users.noreply.github.com> Co-authored-by: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
co-authored by
Jakob Wennberg
Claude Opus 4.8
parent
524d9978f1
commit
704bf93e08
@@ -0,0 +1,91 @@
|
||||
/**
|
||||
* Fit and report the auto-booking confidence calibration (RIP-4 step 4).
|
||||
*
|
||||
* READ-ONLY. Reads categorize_calibration_samples, prints the reliability
|
||||
* diagram + expected calibration error, fits an isotonic calibrator, and shows
|
||||
* what the auto-book / suggest / review bands would look like on the calibrated
|
||||
* probability. Run this once real outcomes have accumulated (>= a few hundred);
|
||||
* it changes nothing on its own.
|
||||
*
|
||||
* npx tsx scripts/fit-categorize-calibration.ts
|
||||
*
|
||||
* Note: .env.local points at production; this only SELECTs, so it is safe, but
|
||||
* it is still the prod corpus you are reading.
|
||||
*/
|
||||
import { createClient } from '@supabase/supabase-js'
|
||||
import {
|
||||
reliabilityByBucket,
|
||||
expectedCalibrationError,
|
||||
fitIsotonic,
|
||||
calibrate,
|
||||
bandFor,
|
||||
type Sample,
|
||||
} from '@/lib/agent/categorize/calibration'
|
||||
|
||||
const url = process.env.NEXT_PUBLIC_SUPABASE_URL
|
||||
const key = process.env.SUPABASE_SERVICE_ROLE_KEY
|
||||
if (!url || !key) {
|
||||
console.error('Missing NEXT_PUBLIC_SUPABASE_URL or SUPABASE_SERVICE_ROLE_KEY in .env.local')
|
||||
process.exit(1)
|
||||
}
|
||||
const supabase = createClient(url, key)
|
||||
|
||||
async function main() {
|
||||
const rows: { confidence: number; was_correct: boolean }[] = []
|
||||
const PAGE = 1000
|
||||
for (let from = 0; ; from += PAGE) {
|
||||
const { data, error } = await supabase
|
||||
.from('categorize_calibration_samples')
|
||||
.select('confidence, was_correct')
|
||||
.order('created_at', { ascending: false })
|
||||
.range(from, from + PAGE - 1)
|
||||
if (error) throw error
|
||||
if (!data || data.length === 0) break
|
||||
rows.push(...(data as { confidence: number; was_correct: boolean }[]))
|
||||
if (data.length < PAGE) break
|
||||
}
|
||||
|
||||
const samples: Sample[] = rows.map((r) => ({ confidence: Number(r.confidence), correct: r.was_correct }))
|
||||
console.log(`\nSamples: ${samples.length}`)
|
||||
if (samples.length === 0) {
|
||||
console.log('No calibration samples yet. Let people book AI proposals first.')
|
||||
return
|
||||
}
|
||||
|
||||
const overall = samples.filter((s) => s.correct).length / samples.length
|
||||
console.log(`Overall accuracy (proposal booked unedited): ${(overall * 100).toFixed(1)}%`)
|
||||
console.log(`Expected calibration error (ECE): ${expectedCalibrationError(samples).toFixed(4)}\n`)
|
||||
|
||||
console.log('Reliability diagram (raw confidence bucket → empirical accuracy):')
|
||||
for (const b of reliabilityByBucket(samples)) {
|
||||
if (b.n === 0) continue
|
||||
const bar = '#'.repeat(Math.round(b.accuracy * 20))
|
||||
console.log(
|
||||
` ${b.lo.toFixed(1)}-${b.hi.toFixed(1)} n=${String(b.n).padStart(5)} ` +
|
||||
`conf=${b.meanConfidence.toFixed(2)} acc=${b.accuracy.toFixed(2)} ${bar}`,
|
||||
)
|
||||
}
|
||||
|
||||
const cal = fitIsotonic(samples)
|
||||
if (!cal) {
|
||||
console.log(`\nNot enough data to fit a calibrator yet (need >= 200). Bands stay uncalibrated (no auto-book).`)
|
||||
return
|
||||
}
|
||||
|
||||
console.log(`\nFitted isotonic calibrator on ${cal.fittedOn} samples.`)
|
||||
console.log('Raw → calibrated (and the band for a small, routine amount):')
|
||||
for (const raw of [0.3, 0.5, 0.6, 0.7, 0.8, 0.9, 0.95, 0.99]) {
|
||||
const p = calibrate(raw, cal)
|
||||
const band = bandFor(raw, cal, { amount: 499 })
|
||||
console.log(` ${raw.toFixed(2)} → ${p.toFixed(2)} ${band}`)
|
||||
}
|
||||
console.log(
|
||||
`\nNext: store this calibrator (or its thresholds) where bandFor reads it, ` +
|
||||
`then enable auto-book for the top band once the empirical accuracy there is acceptable.`,
|
||||
)
|
||||
}
|
||||
|
||||
main().catch((e) => {
|
||||
console.error(e)
|
||||
process.exit(1)
|
||||
})
|
||||
Reference in New Issue
Block a user