@cogitator-ai/evals 0.11.0 → 0.12.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +84 -14
- package/dist/calibrate-judge.d.ts +53 -0
- package/dist/calibrate-judge.d.ts.map +1 -0
- package/dist/calibrate-judge.js +114 -0
- package/dist/calibrate-judge.js.map +1 -0
- package/dist/datasets/dataset.d.ts.map +1 -1
- package/dist/datasets/dataset.js +3 -27
- package/dist/datasets/dataset.js.map +1 -1
- package/dist/eval-comparison.d.ts +71 -2
- package/dist/eval-comparison.d.ts.map +1 -1
- package/dist/eval-comparison.js +149 -42
- package/dist/eval-comparison.js.map +1 -1
- package/dist/index.d.ts +16 -4
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +8 -1
- package/dist/index.js.map +1 -1
- package/dist/metrics/g-eval.d.ts +29 -0
- package/dist/metrics/g-eval.d.ts.map +1 -0
- package/dist/metrics/g-eval.js +89 -0
- package/dist/metrics/g-eval.js.map +1 -0
- package/dist/metrics/index.d.ts +3 -1
- package/dist/metrics/index.d.ts.map +1 -1
- package/dist/metrics/index.js +1 -0
- package/dist/metrics/index.js.map +1 -1
- package/dist/metrics/judge-prompt.d.ts +31 -0
- package/dist/metrics/judge-prompt.d.ts.map +1 -0
- package/dist/metrics/judge-prompt.js +90 -0
- package/dist/metrics/judge-prompt.js.map +1 -0
- package/dist/metrics/llm-judge.d.ts +96 -9
- package/dist/metrics/llm-judge.d.ts.map +1 -1
- package/dist/metrics/llm-judge.js +265 -120
- package/dist/metrics/llm-judge.js.map +1 -1
- package/dist/options.d.ts +4 -0
- package/dist/options.d.ts.map +1 -0
- package/dist/options.js +9 -0
- package/dist/options.js.map +1 -0
- package/dist/pairwise.d.ts +53 -0
- package/dist/pairwise.d.ts.map +1 -0
- package/dist/pairwise.js +118 -0
- package/dist/pairwise.js.map +1 -0
- package/dist/schema.d.ts +16 -0
- package/dist/schema.d.ts.map +1 -1
- package/dist/schema.js +17 -0
- package/dist/schema.js.map +1 -1
- package/dist/stats/bootstrap.d.ts +22 -0
- package/dist/stats/bootstrap.d.ts.map +1 -0
- package/dist/stats/bootstrap.js +34 -0
- package/dist/stats/bootstrap.js.map +1 -0
- package/dist/stats/correlation.d.ts +11 -0
- package/dist/stats/correlation.d.ts.map +1 -0
- package/dist/stats/correlation.js +42 -0
- package/dist/stats/correlation.js.map +1 -0
- package/dist/stats/effect-size.d.ts +7 -0
- package/dist/stats/effect-size.d.ts.map +1 -0
- package/dist/stats/effect-size.js +17 -0
- package/dist/stats/effect-size.js.map +1 -0
- package/dist/stats/holm.d.ts +7 -0
- package/dist/stats/holm.d.ts.map +1 -0
- package/dist/stats/holm.js +21 -0
- package/dist/stats/holm.js.map +1 -0
- package/dist/stats/index.d.ts +10 -2
- package/dist/stats/index.d.ts.map +1 -1
- package/dist/stats/index.js +7 -1
- package/dist/stats/index.js.map +1 -1
- package/dist/stats/mcnemar.d.ts +9 -0
- package/dist/stats/mcnemar.d.ts.map +1 -1
- package/dist/stats/mcnemar.js +27 -13
- package/dist/stats/mcnemar.js.map +1 -1
- package/dist/stats/normal.d.ts +5 -0
- package/dist/stats/normal.d.ts.map +1 -0
- package/dist/stats/normal.js +19 -0
- package/dist/stats/normal.js.map +1 -0
- package/dist/stats/precision.d.ts +12 -0
- package/dist/stats/precision.d.ts.map +1 -0
- package/dist/stats/precision.js +29 -0
- package/dist/stats/precision.js.map +1 -0
- package/dist/stats/random.d.ts +6 -0
- package/dist/stats/random.d.ts.map +1 -0
- package/dist/stats/random.js +25 -0
- package/dist/stats/random.js.map +1 -0
- package/dist/stats/ranks.d.ts +5 -0
- package/dist/stats/ranks.d.ts.map +1 -0
- package/dist/stats/ranks.js +24 -0
- package/dist/stats/ranks.js.map +1 -0
- package/dist/stats/t-test.d.ts +4 -0
- package/dist/stats/t-test.d.ts.map +1 -1
- package/dist/stats/t-test.js +9 -3
- package/dist/stats/t-test.js.map +1 -1
- package/dist/stats/wilcoxon.d.ts +19 -0
- package/dist/stats/wilcoxon.d.ts.map +1 -0
- package/dist/stats/wilcoxon.js +70 -0
- package/dist/stats/wilcoxon.js.map +1 -0
- package/package.json +1 -1
package/README.md
CHANGED
|
@@ -19,13 +19,13 @@ Website docs: [Evals](https://cogitator.app/docs/evals), [Datasets](https://cogi
|
|
|
19
19
|
|
|
20
20
|
- **EvalSuite** — Run datasets against agents or plain functions with configurable concurrency, timeouts, and retries
|
|
21
21
|
- **8 Deterministic Metrics** - exactMatch, contains, containsAll, containsAny, similarity, tokenF1, regex, jsonSchema (Zod)
|
|
22
|
-
- **
|
|
22
|
+
- **7 LLM-as-Judge Metrics** - faithfulness, relevance, coherence, helpfulness, contextRelevance, gEval, custom llmMetric with a free prompt or a discrete rubric. Prompts keep every untrusted value in escaped tags, `judge.samples` averages several verdicts, and `calibrateJudge()` checks a judge against labels
|
|
23
23
|
- **4 Statistical Metrics** - latency, cost, tokenUsage with full percentile breakdowns, and passAtK over trials
|
|
24
24
|
- **Custom Metrics** — `metric()` factory for anything domain-specific
|
|
25
25
|
- **Agent Evaluation** - `toolCalls()` and `trajectory()` against expected tool calls, `retrievalRecall()`, `retrievalPrecision()`, `retrievalMRR()` and the `contextRelevance()` judge against expected passages, multi-turn cases on one conversation
|
|
26
26
|
- **Trials and pass@k** - run each case several times and score how often it passes with `passAtK()`
|
|
27
27
|
- **Assertions** — threshold, noRegression, custom assertion with auto-detection of lower-is-better metrics
|
|
28
|
-
- **A/B Testing**
|
|
28
|
+
- **A/B Testing** - EvalComparison with McNemar's test (exact for few discordant pairs), the paired t-test or Wilcoxon signed-rank, Holm's correction across metrics, bootstrap confidence intervals, effect sizes, latency and cost comparisons and pairwise judging
|
|
29
29
|
- **6 Reporters** - console (colored table), JSON, CSV, JUnit XML, GitHub Actions annotations and job summary, CI (exit code on failure)
|
|
30
30
|
- **Vitest** - `describeEval` from `@cogitator-ai/evals/vitest` runs a suite as a test file, one test per case plus one for the assertions, and `evalMatchers` from `@cogitator-ai/evals/matchers` (Vitest 5) adds `toHaveCalledTool`, `toFollowTrajectory`, `toHaveRetrieved` and `toScoreAtLeast`
|
|
31
31
|
- **Builder API** — Fluent `EvalBuilder` for composable eval pipelines
|
|
@@ -235,7 +235,8 @@ Metrics scored by an LLM judge (0.0 to 1.0). They require a `judge` config on th
|
|
|
235
235
|
| `coherence` | Logical structure and readability |
|
|
236
236
|
| `helpfulness` | Practical usefulness to the user |
|
|
237
237
|
| `contextRelevance` | Whether the retrieved passages fit the input |
|
|
238
|
-
| `
|
|
238
|
+
| `gEval` | G-Eval: evaluation steps, then a scale |
|
|
239
|
+
| `llmMetric` | Custom prompt or rubric, your criteria |
|
|
239
240
|
|
|
240
241
|
```typescript
|
|
241
242
|
import { faithfulness, relevance, llmMetric } from '@cogitator-ai/evals';
|
|
@@ -263,7 +264,43 @@ Dataset.from([{ input: 'Write the lede', context: { dossier: dossierText } }]);
|
|
|
263
264
|
|
|
264
265
|
When the run retrieved passages (see [Retrieval](#retrieval)), `faithfulness()` checks the response against them instead of `context`. `contextRelevance()` judges the input against the retrieved passages alone, or against `context` when nothing was retrieved, and scores 0 without a judge call when the case has neither.
|
|
265
266
|
|
|
266
|
-
The `judge` option is a `JudgeConfigInput`: `{ model, temperature?, maxTokens?, cogitator?, cache?, concurrency? }`, with `temperature` defaulting to `0`. `cache: true` reuses a verdict for the same model, settings and prompt in memory (a `JudgeCache` with `get` and `set` keeps them across runs), and `concurrency` caps the judge calls running at once.
|
|
267
|
+
The `judge` option is a `JudgeConfigInput`: `{ model, temperature?, maxTokens?, cogitator?, cache?, concurrency?, samples? }`, with `temperature` defaulting to `0`. `cache: true` reuses a verdict for the same model, settings and prompt in memory (a `JudgeCache` with `get` and `set` keeps them across runs), and `concurrency` caps the judge calls running at once.
|
|
268
|
+
|
|
269
|
+
`llmMetric` also takes a discrete `rubric` instead of a free prompt: the judge picks one level, and the score is normalized to 0 - 1 by the lowest and highest level. A rubric needs at least two levels with distinct scores, and a verdict that is not a level counts as a judge error.
|
|
270
|
+
|
|
271
|
+
```typescript
|
|
272
|
+
llmMetric({
|
|
273
|
+
name: 'correctness',
|
|
274
|
+
prompt: 'Grade whether the response answers the question correctly.', // optional with a rubric
|
|
275
|
+
rubric: [
|
|
276
|
+
{ score: 1, description: 'wrong, or does not answer' },
|
|
277
|
+
{ score: 3, description: 'partly correct' },
|
|
278
|
+
{ score: 5, description: 'fully correct' },
|
|
279
|
+
],
|
|
280
|
+
});
|
|
281
|
+
```
|
|
282
|
+
|
|
283
|
+
`gEval({ name, criteria, steps?, scale? })` follows G-Eval: the judge scores each case on `scale` (`[1, 5]` by default, normalized to 0 - 1) following evaluation steps. Without `steps`, the judge writes them from `criteria` once per metric instance, before the first case, and every later case reuses them. A failed generation scores the cases waiting on it as judge errors, and the next case tries again. The original G-Eval weights each score by the probability of its token, which not every provider exposes, so set `judge.samples` (with a `temperature` above 0) to average several verdicts instead.
|
|
284
|
+
|
|
285
|
+
```typescript
|
|
286
|
+
gEval({
|
|
287
|
+
name: 'clarity',
|
|
288
|
+
criteria: 'A beginner can follow the explanation, and it names the key idea first.',
|
|
289
|
+
});
|
|
290
|
+
```
|
|
291
|
+
|
|
292
|
+
Every judge prompt keeps the untrusted values of a case (input, response, expected answer, context, retrieved passages, the turns of a conversation) inside XML-like tags such as `<input>` and `<response>`, with `<`, `>` and `&` escaped, so an answer cannot close its tag or open another, and the judge is told that tag contents are data to evaluate, never instructions. An answer that tells the judge to give it full marks is judged as content.
|
|
293
|
+
|
|
294
|
+
`judge.samples` asks the judge that many times per case and metric (1 by default) and scores the mean of the verdicts it could read. The score's `details` give the spread (`mean of 3 verdicts, spread 0.6..0.9`), `metadata.samples` each verdict, and `usage` the summed cost. With `cache`, each sample is cached under its own key.
|
|
295
|
+
|
|
296
|
+
`calibrateJudge(metric, examples, options?)` checks a judge against labels people gave: it scores each `{ input, output, expected?, context?, label }` (labels from 0 to 1) and returns `{ n, errors, pearson, spearman, mae, kappa, perCase }`, `kappa` being Cohen's kappa of pass and fail at `threshold` (0.5 by default). Pass a metric bound with `bindJudgeContext`, or `judge` with a `cogitator` to bind it. Examples the metric fails on are reported in `perCase` and left out of the figures.
|
|
297
|
+
|
|
298
|
+
```typescript
|
|
299
|
+
const calibration = await calibrateJudge(correctness, labelled, {
|
|
300
|
+
judge: { model: 'openai/gpt-6.1-sol', cogitator: cog },
|
|
301
|
+
});
|
|
302
|
+
console.log(calibration.spearman, calibration.kappa);
|
|
303
|
+
```
|
|
267
304
|
|
|
268
305
|
The judge answers with `{ "score", "reasoning" }` JSON. A verdict in a code fence or with the score as a numeric string is read as well, and an answer without JSON falls back to its last labelled score (`Score: 0.4`, `score is 4/5`), never to a stray number in the prose. When no verdict can be read, or the judge call fails, the metric scores 0 with the reason in `score.error`. The judge's usage is reported on each score and counted in `stats.judgeCost`.
|
|
269
306
|
|
|
@@ -505,6 +542,30 @@ for (const [name, mc] of Object.entries(result.summary.metrics)) {
|
|
|
505
542
|
}
|
|
506
543
|
```
|
|
507
544
|
|
|
545
|
+
Binary metrics (every score 0 or 1) take McNemar's test, exact below 25 discordant pairs and chi-square with continuity correction from 25. Other metrics take the paired t-test, or Wilcoxon's signed-rank test with `test: 'wilcoxon'` (zero differences dropped, exact up to 20 pairs, normal approximation above). Paired differences are rounded to 12 significant digits, so float noise neither splits ties nor hides a zero. The winner follows the test's direction: the higher mean for McNemar's and the t-test, the larger rank sum for Wilcoxon's, which can disagree with `difference`, the mean difference. The p-values are corrected with Holm's method across the compared metrics: `pValue` is the test's own, `adjustedPValue` the corrected one, and `significant` means `adjustedPValue < alpha` (`alpha` 0.05 by default). Each metric also reports `difference` (the mean of challenger minus baseline), `ci` (a percentile bootstrap interval of the paired differences at `1 - alpha`, the same on every run with `seed`), `effectSize` (Cohen's d of the paired differences) and the `test` it ran.
|
|
546
|
+
|
|
547
|
+
`result.summary.latency` and `result.summary.cost` compare milliseconds and USD per case, paired over the cases both sides answered, where lower is better: means, `difference`, `pValue` and `winner`. `cost` is there only when both sides reported usage. Neither takes part in the metrics' vote on the overall winner.
|
|
548
|
+
|
|
549
|
+
With `pairwise`, the judge also reads both answers of each case and says which is better, twice, the second time with the answers swapped. Agreeing verdicts give the case to that side, disagreeing ones or a tie make it a tie, a side that errored loses the case without a judge call, and a case both sides errored on is left out. A judge failure counts as a tie and is reported.
|
|
550
|
+
|
|
551
|
+
```typescript
|
|
552
|
+
const comparison = new EvalComparison({
|
|
553
|
+
dataset,
|
|
554
|
+
targets: { baseline, challenger },
|
|
555
|
+
metrics: [exactMatch()],
|
|
556
|
+
judge: { model: 'openai/gpt-6.1-sol', cogitator: cog },
|
|
557
|
+
pairwise: { criteria: 'Which answer is more accurate.' },
|
|
558
|
+
test: 'wilcoxon',
|
|
559
|
+
alpha: 0.05,
|
|
560
|
+
seed: 42,
|
|
561
|
+
});
|
|
562
|
+
|
|
563
|
+
const { pairwise } = await comparison.run();
|
|
564
|
+
// { name, wins: { baseline, challenger }, ties, cases, winRate, judgeErrors, cost, perCase }
|
|
565
|
+
```
|
|
566
|
+
|
|
567
|
+
`winRate` is the challenger's wins over the decided cases, NaN when none was decided.
|
|
568
|
+
|
|
508
569
|
Access full suite results via `result.baseline` and `result.challenger`. `comparison.run({ signal })` cancels both runs.
|
|
509
570
|
|
|
510
571
|
---
|
|
@@ -657,6 +718,7 @@ result.report('ci');
|
|
|
657
718
|
| ---------------- | ------------------------------------------------------------------- |
|
|
658
719
|
| `EvalSuite` | Main evaluation runner |
|
|
659
720
|
| `EvalComparison` | A/B testing runner with statistical significance |
|
|
721
|
+
| `calibrateJudge` | Agreement of a judge metric with labelled examples |
|
|
660
722
|
| `EvalBuilder` | Fluent builder for EvalSuite |
|
|
661
723
|
| `Dataset` | Immutable dataset with from/fromJsonl/fromCsv/filter/sample/shuffle |
|
|
662
724
|
| `loadJsonl` | Low-level JSONL file loader |
|
|
@@ -680,7 +742,8 @@ result.report('ci');
|
|
|
680
742
|
| `coherence` | LLM Judge | Logical structure |
|
|
681
743
|
| `helpfulness` | LLM Judge | Practical usefulness |
|
|
682
744
|
| `contextRelevance` | LLM Judge | Relevance of the retrieved passages to the input |
|
|
683
|
-
| `
|
|
745
|
+
| `gEval` | LLM Judge | Evaluation steps from criteria, then a score on a scale |
|
|
746
|
+
| `llmMetric` | LLM Judge | Custom judge prompt or discrete rubric |
|
|
684
747
|
| `toolCalls` | Agent | Tool calls against `expectedToolCalls`, F1, recall or precision |
|
|
685
748
|
| `trajectory` | Agent | Sequence of tool names, exact, in-order or any-order |
|
|
686
749
|
| `retrievalRecall` | Agent | Share of `expectedPassages` retrieved |
|
|
@@ -726,15 +789,22 @@ result.report('ci');
|
|
|
726
789
|
|
|
727
790
|
### Statistics
|
|
728
791
|
|
|
729
|
-
| Export
|
|
730
|
-
|
|
|
731
|
-
| `pairedTTest`
|
|
732
|
-
| `
|
|
733
|
-
| `
|
|
734
|
-
| `
|
|
735
|
-
| `
|
|
736
|
-
| `
|
|
737
|
-
| `
|
|
792
|
+
| Export | Description |
|
|
793
|
+
| -------------------- | --------------------------------------------------------------- |
|
|
794
|
+
| `pairedTTest` | Paired t-test for continuous metric comparison |
|
|
795
|
+
| `wilcoxonSignedRank` | Wilcoxon signed-rank test, zeros dropped, exact up to 20 pairs |
|
|
796
|
+
| `mcnemarsTest` | McNemar's test (chi-square, continuity corrected) for binary |
|
|
797
|
+
| `mcnemarExact` | Exact McNemar's test, the binomial test of the discordant pairs |
|
|
798
|
+
| `holm` | Holm-Bonferroni adjusted p-values |
|
|
799
|
+
| `bootstrapCI` | Seeded percentile bootstrap confidence interval of the mean |
|
|
800
|
+
| `cohensD` | Cohen's d of paired differences (dz) |
|
|
801
|
+
| `pearson` | Pearson's correlation |
|
|
802
|
+
| `spearman` | Spearman's rank correlation |
|
|
803
|
+
| `mean` | Arithmetic mean |
|
|
804
|
+
| `median` | Median value |
|
|
805
|
+
| `stdDev` | Sample standard deviation |
|
|
806
|
+
| `percentile` | Arbitrary percentile |
|
|
807
|
+
| `aggregate` | Full stats: mean, median, min, max, stdDev, p50, p95, p99 |
|
|
738
808
|
|
|
739
809
|
### Agent Tools
|
|
740
810
|
|
|
@@ -0,0 +1,53 @@
|
|
|
1
|
+
import type { MetricFn } from './metrics/types.js';
|
|
2
|
+
import type { JudgeConfigInput } from './schema.js';
|
|
3
|
+
/** An answer labelled by a person, with the score the metric should give it, from 0 to 1 */
|
|
4
|
+
export interface CalibrationExample {
|
|
5
|
+
input: string;
|
|
6
|
+
output: string;
|
|
7
|
+
expected?: string;
|
|
8
|
+
context?: Record<string, unknown>;
|
|
9
|
+
label: number;
|
|
10
|
+
}
|
|
11
|
+
export interface CalibrateJudgeOptions {
|
|
12
|
+
/** Where a score or label counts as a pass for `kappa`, 0.5 by default */
|
|
13
|
+
threshold?: number;
|
|
14
|
+
/** Binds an unbound judge metric, run by `judge.cogitator` */
|
|
15
|
+
judge?: JudgeConfigInput;
|
|
16
|
+
/** Examples scored at once, 5 by default */
|
|
17
|
+
concurrency?: number;
|
|
18
|
+
}
|
|
19
|
+
export interface CalibrationCase {
|
|
20
|
+
/** Position of the example */
|
|
21
|
+
index: number;
|
|
22
|
+
label: number;
|
|
23
|
+
/** The metric's score, absent when it failed */
|
|
24
|
+
score?: number;
|
|
25
|
+
details?: string;
|
|
26
|
+
error?: string;
|
|
27
|
+
}
|
|
28
|
+
export interface JudgeCalibration {
|
|
29
|
+
/** Examples the metric scored, those it failed on left out of the figures */
|
|
30
|
+
n: number;
|
|
31
|
+
/** Examples the metric failed on */
|
|
32
|
+
errors: number;
|
|
33
|
+
/** Pearson's correlation of scores and labels, NaN when either does not vary */
|
|
34
|
+
pearson: number;
|
|
35
|
+
/** Spearman's rank correlation of scores and labels */
|
|
36
|
+
spearman: number;
|
|
37
|
+
/** Mean absolute error of the scores against the labels */
|
|
38
|
+
mae: number;
|
|
39
|
+
/**
|
|
40
|
+
* Cohen's kappa of pass and fail, a score or label passing at `threshold` or more. NaN when
|
|
41
|
+
* agreement by chance is certain, every score and label being on one side
|
|
42
|
+
*/
|
|
43
|
+
kappa: number;
|
|
44
|
+
perCase: CalibrationCase[];
|
|
45
|
+
}
|
|
46
|
+
/**
|
|
47
|
+
* How well a metric, usually an LLM judge, agrees with labels people gave: it scores each
|
|
48
|
+
* labelled example and compares the scores with the labels. Bind a judge metric first with
|
|
49
|
+
* `bindJudgeContext`, or pass `judge` with a `cogitator`. Examples the metric fails on are
|
|
50
|
+
* reported and left out of the figures
|
|
51
|
+
*/
|
|
52
|
+
export declare function calibrateJudge(metric: MetricFn, examples: readonly CalibrationExample[], options?: CalibrateJudgeOptions): Promise<JudgeCalibration>;
|
|
53
|
+
//# sourceMappingURL=calibrate-judge.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"calibrate-judge.d.ts","sourceRoot":"","sources":["../src/calibrate-judge.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAkB,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAIhE,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAC;AAKjD,4FAA4F;AAC5F,MAAM,WAAW,kBAAkB;IACjC,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAClC,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,qBAAqB;IACpC,0EAA0E;IAC1E,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,8DAA8D;IAC9D,KAAK,CAAC,EAAE,gBAAgB,CAAC;IACzB,4CAA4C;IAC5C,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,eAAe;IAC9B,8BAA8B;IAC9B,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,MAAM,CAAC;IACd,gDAAgD;IAChD,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,gBAAgB;IAC/B,6EAA6E;IAC7E,CAAC,EAAE,MAAM,CAAC;IACV,oCAAoC;IACpC,MAAM,EAAE,MAAM,CAAC;IACf,gFAAgF;IAChF,OAAO,EAAE,MAAM,CAAC;IAChB,uDAAuD;IACvD,QAAQ,EAAE,MAAM,CAAC;IACjB,2DAA2D;IAC3D,GAAG,EAAE,MAAM,CAAC;IACZ;;;OAGG;IACH,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,eAAe,EAAE,CAAC;CAC5B;AA6ED;;;;;GAKG;AACH,wBAAsB,cAAc,CAClC,MAAM,EAAE,QAAQ,EAChB,QAAQ,EAAE,SAAS,kBAAkB,EAAE,EACvC,OAAO,GAAE,qBAA0B,GAClC,OAAO,CAAC,gBAAgB,CAAC,CAyC3B"}
|
|
@@ -0,0 +1,114 @@
|
|
|
1
|
+
import { z } from 'zod';
|
|
2
|
+
import { isLLMMetric } from './eval-suite.js';
|
|
3
|
+
import { nonFiniteScoreError } from './metrics/types.js';
|
|
4
|
+
import { bindJudgeContext, judgeContextFor } from './metrics/llm-judge.js';
|
|
5
|
+
import { JudgeConfigSchema } from './schema.js';
|
|
6
|
+
import { parseOptions } from './options.js';
|
|
7
|
+
import { pearson, spearman } from './stats/correlation.js';
|
|
8
|
+
import { mean } from './stats/percentiles.js';
|
|
9
|
+
const CalibrationExampleSchema = z.object({
|
|
10
|
+
input: z.string(),
|
|
11
|
+
output: z.string(),
|
|
12
|
+
expected: z.string().optional(),
|
|
13
|
+
context: z.record(z.string(), z.unknown()).optional(),
|
|
14
|
+
label: z.number().min(0).max(1),
|
|
15
|
+
});
|
|
16
|
+
const CalibrateJudgeOptionsSchema = z.object({
|
|
17
|
+
threshold: z.number().min(0).max(1).default(0.5),
|
|
18
|
+
concurrency: z.number().int().positive().default(5),
|
|
19
|
+
});
|
|
20
|
+
/** Cohen's kappa of two binary raters */
|
|
21
|
+
function binaryKappa(a, b) {
|
|
22
|
+
const n = a.length;
|
|
23
|
+
if (n === 0)
|
|
24
|
+
return Number.NaN;
|
|
25
|
+
let agree = 0;
|
|
26
|
+
let aPass = 0;
|
|
27
|
+
let bPass = 0;
|
|
28
|
+
for (let i = 0; i < n; i++) {
|
|
29
|
+
if (a[i] === b[i])
|
|
30
|
+
agree++;
|
|
31
|
+
if (a[i])
|
|
32
|
+
aPass++;
|
|
33
|
+
if (b[i])
|
|
34
|
+
bPass++;
|
|
35
|
+
}
|
|
36
|
+
const observed = agree / n;
|
|
37
|
+
const chance = (aPass / n) * (bPass / n) + (1 - aPass / n) * (1 - bPass / n);
|
|
38
|
+
return chance === 1 ? Number.NaN : (observed - chance) / (1 - chance);
|
|
39
|
+
}
|
|
40
|
+
function bindMetric(metric, judge) {
|
|
41
|
+
if (!isLLMMetric(metric))
|
|
42
|
+
return metric;
|
|
43
|
+
if (!judge) {
|
|
44
|
+
throw new Error('calibrateJudge needs options.judge to run an unbound judge metric');
|
|
45
|
+
}
|
|
46
|
+
const config = JudgeConfigSchema.parse(judge);
|
|
47
|
+
if (!config.cogitator) {
|
|
48
|
+
throw new Error('calibrateJudge needs judge.cogitator to run the judge');
|
|
49
|
+
}
|
|
50
|
+
return bindJudgeContext(metric, judgeContextFor(config.cogitator, config));
|
|
51
|
+
}
|
|
52
|
+
async function scoreExample(metric, example, index) {
|
|
53
|
+
const result = {
|
|
54
|
+
case: {
|
|
55
|
+
input: example.input,
|
|
56
|
+
...(example.expected !== undefined && { expected: example.expected }),
|
|
57
|
+
...(example.context !== undefined && { context: example.context }),
|
|
58
|
+
},
|
|
59
|
+
output: example.output,
|
|
60
|
+
duration: 0,
|
|
61
|
+
};
|
|
62
|
+
try {
|
|
63
|
+
const scored = await metric(result);
|
|
64
|
+
const invalid = scored.error ?? nonFiniteScoreError(scored.score);
|
|
65
|
+
if (invalid !== undefined)
|
|
66
|
+
return { index, label: example.label, error: invalid };
|
|
67
|
+
return {
|
|
68
|
+
index,
|
|
69
|
+
label: example.label,
|
|
70
|
+
score: scored.score,
|
|
71
|
+
...(scored.details !== undefined && { details: scored.details }),
|
|
72
|
+
};
|
|
73
|
+
}
|
|
74
|
+
catch (error) {
|
|
75
|
+
return {
|
|
76
|
+
index,
|
|
77
|
+
label: example.label,
|
|
78
|
+
error: error instanceof Error ? error.message : String(error),
|
|
79
|
+
};
|
|
80
|
+
}
|
|
81
|
+
}
|
|
82
|
+
/**
|
|
83
|
+
* How well a metric, usually an LLM judge, agrees with labels people gave: it scores each
|
|
84
|
+
* labelled example and compares the scores with the labels. Bind a judge metric first with
|
|
85
|
+
* `bindJudgeContext`, or pass `judge` with a `cogitator`. Examples the metric fails on are
|
|
86
|
+
* reported and left out of the figures
|
|
87
|
+
*/
|
|
88
|
+
export async function calibrateJudge(metric, examples, options = {}) {
|
|
89
|
+
const { threshold, concurrency } = parseOptions(CalibrateJudgeOptionsSchema, { threshold: options.threshold, concurrency: options.concurrency }, 'calibrateJudge');
|
|
90
|
+
const parsed = parseOptions(z.array(CalibrationExampleSchema), examples, 'calibrateJudge examples');
|
|
91
|
+
const bound = bindMetric(metric, options.judge);
|
|
92
|
+
const perCase = new Array(parsed.length);
|
|
93
|
+
let next = 0;
|
|
94
|
+
const worker = async () => {
|
|
95
|
+
while (next < parsed.length) {
|
|
96
|
+
const at = next++;
|
|
97
|
+
perCase[at] = await scoreExample(bound, parsed[at], at);
|
|
98
|
+
}
|
|
99
|
+
};
|
|
100
|
+
await Promise.all(Array.from({ length: Math.min(concurrency, parsed.length) }, worker));
|
|
101
|
+
const scored = perCase.filter((one) => one.score !== undefined);
|
|
102
|
+
const scores = scored.map((one) => one.score);
|
|
103
|
+
const labels = scored.map((one) => one.label);
|
|
104
|
+
return {
|
|
105
|
+
n: scored.length,
|
|
106
|
+
errors: perCase.length - scored.length,
|
|
107
|
+
pearson: pearson(scores, labels),
|
|
108
|
+
spearman: spearman(scores, labels),
|
|
109
|
+
mae: scored.length === 0 ? Number.NaN : mean(scored.map((one) => Math.abs(one.score - one.label))),
|
|
110
|
+
kappa: binaryKappa(scores.map((score) => score >= threshold), labels.map((label) => label >= threshold)),
|
|
111
|
+
perCase,
|
|
112
|
+
};
|
|
113
|
+
}
|
|
114
|
+
//# sourceMappingURL=calibrate-judge.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"calibrate-judge.js","sourceRoot":"","sources":["../src/calibrate-judge.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAE3C,OAAO,EAAE,mBAAmB,EAAE,MAAM,iBAAiB,CAAC;AACtD,OAAO,EAAE,gBAAgB,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AACxE,OAAO,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAE7C,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,EAAE,IAAI,EAAE,MAAM,qBAAqB,CAAC;AAiD3C,MAAM,wBAAwB,GAAG,CAAC,CAAC,MAAM,CAAC;IACxC,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;IACjB,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;IAClB,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC/B,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,QAAQ,EAAE;IACrD,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;CAChC,CAAC,CAAC;AAEH,MAAM,2BAA2B,GAAG,CAAC,CAAC,MAAM,CAAC;IAC3C,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC;IAChD,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;CACpD,CAAC,CAAC;AAEH,yCAAyC;AACzC,SAAS,WAAW,CAAC,CAAqB,EAAE,CAAqB;IAC/D,MAAM,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC;IACnB,IAAI,CAAC,KAAK,CAAC;QAAE,OAAO,MAAM,CAAC,GAAG,CAAC;IAC/B,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC3B,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YAAE,KAAK,EAAE,CAAC;QAC3B,IAAI,CAAC,CAAC,CAAC,CAAC;YAAE,KAAK,EAAE,CAAC;QAClB,IAAI,CAAC,CAAC,CAAC,CAAC;YAAE,KAAK,EAAE,CAAC;IACpB,CAAC;IACD,MAAM,QAAQ,GAAG,KAAK,GAAG,CAAC,CAAC;IAC3B,MAAM,MAAM,GAAG,CAAC,KAAK,GAAG,CAAC,CAAC,GAAG,CAAC,KAAK,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,CAAC,CAAC;IAC7E,OAAO,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC;AACxE,CAAC;AAED,SAAS,UAAU,CAAC,MAAgB,EAAE,KAAmC;IACvE,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;QAAE,OAAO,MAAM,CAAC;IACxC,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,MAAM,IAAI,KAAK,CAAC,mEAAmE,CAAC,CAAC;IACvF,CAAC;IACD,MAAM,MAAM,GAAG,iBAAiB,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IAC9C,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;QACtB,MAAM,IAAI,KAAK,CAAC,uDAAuD,CAAC,CAAC;IAC3E,CAAC;IACD,OAAO,gBAAgB,CAAC,MAAM,EAAE,eAAe,CAAC,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC;AAC7E,CAAC;AAED,KAAK,UAAU,YAAY,CACzB,MAAgB,EAChB,OAAkD,EAClD,KAAa;IAEb,MAAM,MAAM,GAAmB;QAC7B,IAAI,EAAE;YACJ,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,GAAG,CAAC,OAAO,CAAC,QAAQ,KAAK,SAAS,IAAI,EAAE,QAAQ,EAAE,OAAO,CAAC,QAAQ,EAAE,CAAC;YACrE,GAAG,CAAC,OAAO,CAAC,OAAO,KAAK,SAAS,IAAI,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC;SACnE;QACD,MAAM,EAAE,OAAO,CAAC,MAAM;QACtB,QAAQ,EAAE,CAAC;KACZ,CAAC;IACF,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,CAAC;QACpC,MAAM,OAAO,GAAG,MAAM,CAAC,KAAK,IAAI,mBAAmB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAClE,IAAI,OAAO,KAAK,SAAS;YAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;QAClF,OAAO;YACL,KAAK;YACL,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,GAAG,CAAC,MAAM,CAAC,OAAO,KAAK,SAAS,IAAI,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC;SACjE,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO;YACL,KAAK;YACL,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;SAC9D,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;;;;GAKG;AACH,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,MAAgB,EAChB,QAAuC,EACvC,UAAiC,EAAE;IAEnC,MAAM,EAAE,SAAS,EAAE,WAAW,EAAE,GAAG,YAAY,CAC7C,2BAA2B,EAC3B,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE,WAAW,EAAE,OAAO,CAAC,WAAW,EAAE,EAClE,gBAAgB,CACjB,CAAC;IACF,MAAM,MAAM,GAAG,YAAY,CACzB,CAAC,CAAC,KAAK,CAAC,wBAAwB,CAAC,EACjC,QAAQ,EACR,yBAAyB,CAC1B,CAAC;IACF,MAAM,KAAK,GAAG,UAAU,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;IAEhD,MAAM,OAAO,GAAG,IAAI,KAAK,CAAkB,MAAM,CAAC,MAAM,CAAC,CAAC;IAC1D,IAAI,IAAI,GAAG,CAAC,CAAC;IACb,MAAM,MAAM,GAAG,KAAK,IAAI,EAAE;QACxB,OAAO,IAAI,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC;YAC5B,MAAM,EAAE,GAAG,IAAI,EAAE,CAAC;YAClB,OAAO,CAAC,EAAE,CAAC,GAAG,MAAM,YAAY,CAAC,KAAK,EAAE,MAAM,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC,CAAC;IACF,MAAM,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,MAAM,CAAC,MAAM,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;IAExF,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAC3B,CAAC,GAAG,EAA8C,EAAE,CAAC,GAAG,CAAC,KAAK,KAAK,SAAS,CAC7E,CAAC;IACF,MAAM,MAAM,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAC9C,MAAM,MAAM,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAC9C,OAAO;QACL,CAAC,EAAE,MAAM,CAAC,MAAM;QAChB,MAAM,EAAE,OAAO,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM;QACtC,OAAO,EAAE,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC;QAChC,QAAQ,EAAE,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;QAClC,GAAG,EACD,MAAM,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC;QAC/F,KAAK,EAAE,WAAW,CAChB,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,IAAI,SAAS,CAAC,EACzC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,IAAI,SAAS,CAAC,CAC1C;QACD,OAAO;KACR,CAAC;AACJ,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"dataset.d.ts","sourceRoot":"","sources":["../../src/datasets/dataset.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,WAAW,CAAC;
|
|
1
|
+
{"version":3,"file":"dataset.d.ts","sourceRoot":"","sources":["../../src/datasets/dataset.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,WAAW,CAAC;AAK1C,qBAAa,OAAO;IAClB,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAsB;IAE7C,OAAO;IAIP,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,QAAQ,EAAE,GAAG,OAAO;WAI1B,SAAS,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;WAKzC,OAAO,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAKpD,IAAI,MAAM,IAAI,MAAM,CAEnB;IAED,IAAI,KAAK,IAAI,SAAS,QAAQ,EAAE,CAE/B;IAED,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,QAAQ,CAAC,QAAQ,CAAC;IAavC,MAAM,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,QAAQ,KAAK,OAAO,GAAG,OAAO;IAI7C;;;OAGG;IACH,MAAM,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,GAAE;QAAE,IAAI,CAAC,EAAE,MAAM,GAAG,MAAM,CAAA;KAAO,GAAG,OAAO;IAYpE,yEAAyE;IACzE,OAAO,CAAC,OAAO,GAAE;QAAE,IAAI,CAAC,EAAE,MAAM,GAAG,MAAM,CAAA;KAAO,GAAG,OAAO;CAG3D"}
|
package/dist/datasets/dataset.js
CHANGED
|
@@ -1,6 +1,7 @@
|
|
|
1
1
|
import { EvalCaseSchema } from '../schema.js';
|
|
2
2
|
import { loadJsonl } from './jsonl-loader.js';
|
|
3
3
|
import { loadCsv } from './csv-loader.js';
|
|
4
|
+
import { seededRandom } from '../stats/random.js';
|
|
4
5
|
export class Dataset {
|
|
5
6
|
_cases;
|
|
6
7
|
constructor(cases) {
|
|
@@ -47,12 +48,12 @@ export class Dataset {
|
|
|
47
48
|
return new Dataset([]);
|
|
48
49
|
}
|
|
49
50
|
const count = Math.min(n, this._cases.length);
|
|
50
|
-
const indices = shuffled(Array.from({ length: this._cases.length }, (_, i) => i),
|
|
51
|
+
const indices = shuffled(Array.from({ length: this._cases.length }, (_, i) => i), seededRandom(options.seed));
|
|
51
52
|
return new Dataset(indices.slice(0, count).map((i) => this._cases[i]));
|
|
52
53
|
}
|
|
53
54
|
/** The cases in random order, the same order on every run with `seed` */
|
|
54
55
|
shuffle(options = {}) {
|
|
55
|
-
return new Dataset(shuffled([...this._cases],
|
|
56
|
+
return new Dataset(shuffled([...this._cases], seededRandom(options.seed)));
|
|
56
57
|
}
|
|
57
58
|
}
|
|
58
59
|
/** Shuffles `items` in place (Fisher-Yates) with the random numbers `random` gives */
|
|
@@ -63,29 +64,4 @@ function shuffled(items, random) {
|
|
|
63
64
|
}
|
|
64
65
|
return items;
|
|
65
66
|
}
|
|
66
|
-
/**
|
|
67
|
-
* `Math.random`, or a mulberry32 generator seeded by the hash of `seed`, so every number and
|
|
68
|
-
* string seeds its own sequence (`1.2` and `1.7`, `-1` and `2 ** 32 - 1` alike)
|
|
69
|
-
*/
|
|
70
|
-
function randomOf(seed) {
|
|
71
|
-
if (seed === undefined)
|
|
72
|
-
return Math.random;
|
|
73
|
-
let state = hashSeed(String(seed));
|
|
74
|
-
return () => {
|
|
75
|
-
state = (state + 0x6d2b79f5) >>> 0;
|
|
76
|
-
let t = state;
|
|
77
|
-
t = Math.imul(t ^ (t >>> 15), t | 1);
|
|
78
|
-
t ^= t + Math.imul(t ^ (t >>> 7), t | 61);
|
|
79
|
-
return ((t ^ (t >>> 14)) >>> 0) / 4294967296;
|
|
80
|
-
};
|
|
81
|
-
}
|
|
82
|
-
/** FNV-1a of a string, to seed the generator */
|
|
83
|
-
function hashSeed(seed) {
|
|
84
|
-
let hash = 0x811c9dc5;
|
|
85
|
-
for (let i = 0; i < seed.length; i++) {
|
|
86
|
-
hash ^= seed.charCodeAt(i);
|
|
87
|
-
hash = Math.imul(hash, 0x01000193) >>> 0;
|
|
88
|
-
}
|
|
89
|
-
return hash;
|
|
90
|
-
}
|
|
91
67
|
//# sourceMappingURL=dataset.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"dataset.js","sourceRoot":"","sources":["../../src/datasets/dataset.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAE3C,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAC3C,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;
|
|
1
|
+
{"version":3,"file":"dataset.js","sourceRoot":"","sources":["../../src/datasets/dataset.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAE3C,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAC3C,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AACvC,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAE/C,MAAM,OAAO,OAAO;IACD,MAAM,CAAsB;IAE7C,YAAoB,KAAiB;QACnC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACzE,CAAC;IAED,MAAM,CAAC,IAAI,CAAC,KAAiB;QAC3B,OAAO,IAAI,OAAO,CAAC,KAAK,CAAC,CAAC;IAC5B,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,IAAY;QACjC,MAAM,KAAK,GAAG,MAAM,SAAS,CAAC,IAAI,CAAC,CAAC;QACpC,OAAO,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC7B,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,IAAY;QAC/B,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;QAClC,OAAO,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC7B,CAAC;IAED,IAAI,MAAM;QACR,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC;IAC5B,CAAC;IAED,IAAI,KAAK;QACP,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED,CAAC,MAAM,CAAC,QAAQ,CAAC;QACf,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC;QAC1B,OAAO;YACL,IAAI;gBACF,IAAI,KAAK,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;oBACzB,OAAO,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;gBAChD,CAAC;gBACD,OAAO,EAAE,KAAK,EAAE,SAAkB,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;YACnD,CAAC;SACF,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,EAA4B;QACjC,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IAC7C,CAAC;IAED;;;OAGG;IACH,MAAM,CAAC,CAAS,EAAE,UAAsC,EAAE;QACxD,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvC,OAAO,IAAI,OAAO,CAAC,EAAE,CAAC,CAAC;QACzB,CAAC;QACD,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC9C,MAAM,OAAO,GAAG,QAAQ,CACtB,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EACvD,YAAY,CAAC,OAAO,CAAC,IAAI,CAAC,CAC3B,CAAC;QACF,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACzE,CAAC;IAED,yEAAyE;IACzE,OAAO,CAAC,UAAsC,EAAE;QAC9C,OAAO,IAAI,OAAO,CAAC,QAAQ,CAAC,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,EAAE,YAAY,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC7E,CAAC;CACF;AAED,sFAAsF;AACtF,SAAS,QAAQ,CAAI,KAAU,EAAE,MAAoB;IACnD,KAAK,IAAI,CAAC,GAAG,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC1C,MAAM,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;QACzC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAE,EAAE,KAAK,CAAC,CAAC,CAAE,CAAC,CAAC;IAChD,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC"}
|
|
@@ -2,6 +2,7 @@ import type { EvalTarget, EvalSuiteResult } from './eval-suite.js';
|
|
|
2
2
|
import type { MetricFn, StatisticalMetricFn } from './metrics/types.js';
|
|
3
3
|
import type { JudgeConfigInput } from './schema.js';
|
|
4
4
|
import { Dataset } from './datasets/index.js';
|
|
5
|
+
import type { PairwiseOptions, PairwiseResult } from './pairwise.js';
|
|
5
6
|
export interface EvalComparisonOptions {
|
|
6
7
|
dataset: Dataset;
|
|
7
8
|
targets: {
|
|
@@ -10,6 +11,7 @@ export interface EvalComparisonOptions {
|
|
|
10
11
|
};
|
|
11
12
|
metrics?: MetricFn[];
|
|
12
13
|
statisticalMetrics?: StatisticalMetricFn[];
|
|
14
|
+
/** The judge of LLM metrics and of `pairwise`, on both sides */
|
|
13
15
|
judge?: JudgeConfigInput;
|
|
14
16
|
concurrency?: number;
|
|
15
17
|
timeout?: number;
|
|
@@ -19,6 +21,23 @@ export interface EvalComparisonOptions {
|
|
|
19
21
|
* mean over its scored trials, and `passAtK()` statistical metrics see every trial
|
|
20
22
|
*/
|
|
21
23
|
trials?: number;
|
|
24
|
+
/**
|
|
25
|
+
* Significance level, 0.05 by default. A metric is significant when its p-value, adjusted with
|
|
26
|
+
* Holm's correction across the compared metrics, is below it, and its `ci` covers `1 - alpha`
|
|
27
|
+
*/
|
|
28
|
+
alpha?: number;
|
|
29
|
+
/**
|
|
30
|
+
* Test of metrics that are not binary, and of latency and cost: `t` (paired t-test, the
|
|
31
|
+
* default) or `wilcoxon` (Wilcoxon signed-rank). Binary metrics always take McNemar's test
|
|
32
|
+
*/
|
|
33
|
+
test?: 't' | 'wilcoxon';
|
|
34
|
+
/** Seeds the bootstrap of each metric's `ci`, so the intervals are the same on every run */
|
|
35
|
+
seed?: number | string;
|
|
36
|
+
/**
|
|
37
|
+
* Asks the judge which side answered each case better, both ways round, into
|
|
38
|
+
* `result.pairwise`. Needs `judge`, run by `judge.cogitator` or an agent target's `cogitator`
|
|
39
|
+
*/
|
|
40
|
+
pairwise?: PairwiseOptions;
|
|
22
41
|
onProgress?: (progress: {
|
|
23
42
|
target: string;
|
|
24
43
|
completed: number;
|
|
@@ -29,25 +48,75 @@ export interface EvalComparisonRunOptions {
|
|
|
29
48
|
/** Cancels both runs; `run()` rejects with the signal's reason */
|
|
30
49
|
signal?: AbortSignal;
|
|
31
50
|
}
|
|
51
|
+
/**
|
|
52
|
+
* The test a comparison ran: McNemar's (exact below 25 discordant pairs, chi-square with
|
|
53
|
+
* continuity correction from 25), the paired t-test, Wilcoxon signed-rank, or none with fewer
|
|
54
|
+
* than 2 cases
|
|
55
|
+
*/
|
|
56
|
+
export type ComparisonTest = 'mcnemar' | 'mcnemar-exact' | 't' | 'wilcoxon' | 'none';
|
|
57
|
+
export type ComparisonWinner = 'baseline' | 'challenger' | 'tie';
|
|
32
58
|
export interface MetricComparison {
|
|
33
59
|
baseline: number;
|
|
34
60
|
challenger: number;
|
|
35
61
|
/** Cases compared: those at least one side scored, an errored side counting as a 0 */
|
|
36
62
|
n: number;
|
|
63
|
+
/** Mean of the paired differences, challenger minus baseline */
|
|
64
|
+
difference: number;
|
|
65
|
+
/** Percentile bootstrap confidence interval of `difference` at `1 - alpha` */
|
|
66
|
+
ci: {
|
|
67
|
+
low: number;
|
|
68
|
+
high: number;
|
|
69
|
+
};
|
|
70
|
+
/** Cohen's d of the paired differences (dz), NaN with fewer than 2 cases */
|
|
71
|
+
effectSize: number;
|
|
72
|
+
test: ComparisonTest;
|
|
73
|
+
/** The test's own p-value */
|
|
74
|
+
pValue: number;
|
|
75
|
+
/** `pValue` after Holm's correction across the compared metrics */
|
|
76
|
+
adjustedPValue: number;
|
|
77
|
+
/** `adjustedPValue` is below `alpha` */
|
|
78
|
+
significant: boolean;
|
|
79
|
+
/**
|
|
80
|
+
* The higher side when significant, by the test's direction: the sign of `difference`, except
|
|
81
|
+
* for Wilcoxon's test, where it is the side with the larger rank sum and may differ from it
|
|
82
|
+
*/
|
|
83
|
+
winner: ComparisonWinner;
|
|
84
|
+
}
|
|
85
|
+
/**
|
|
86
|
+
* Latency or cost of the two sides, paired over the cases both answered, where lower is better.
|
|
87
|
+
* Not corrected for multiple comparisons and not part of the metrics' vote on the winner
|
|
88
|
+
*/
|
|
89
|
+
export interface ResourceComparison {
|
|
90
|
+
/** Mean per case, NaN when no case pairs up */
|
|
91
|
+
baseline: number;
|
|
92
|
+
challenger: number;
|
|
93
|
+
n: number;
|
|
94
|
+
/** Challenger minus baseline: below 0 when the challenger is cheaper or faster */
|
|
95
|
+
difference: number;
|
|
96
|
+
test: ComparisonTest;
|
|
37
97
|
pValue: number;
|
|
38
98
|
significant: boolean;
|
|
39
|
-
|
|
99
|
+
/** The lower side when significant, by the test's direction as for `MetricComparison.winner` */
|
|
100
|
+
winner: ComparisonWinner;
|
|
40
101
|
}
|
|
41
102
|
export interface ComparisonResult {
|
|
42
103
|
summary: {
|
|
43
|
-
winner:
|
|
104
|
+
winner: ComparisonWinner;
|
|
44
105
|
metrics: Record<string, MetricComparison>;
|
|
106
|
+
/** Milliseconds per case, the mean over the answered trials of a case */
|
|
107
|
+
latency: ResourceComparison;
|
|
108
|
+
/** USD per case, over the cases both sides reported usage on. Absent when none did */
|
|
109
|
+
cost?: ResourceComparison;
|
|
45
110
|
};
|
|
46
111
|
baseline: EvalSuiteResult;
|
|
47
112
|
challenger: EvalSuiteResult;
|
|
113
|
+
/** The judge's pairwise verdicts, with the `pairwise` option */
|
|
114
|
+
pairwise?: PairwiseResult;
|
|
48
115
|
}
|
|
49
116
|
export declare class EvalComparison {
|
|
50
117
|
private readonly opts;
|
|
118
|
+
private readonly config;
|
|
119
|
+
private readonly pairwiseJudge?;
|
|
51
120
|
constructor(opts: EvalComparisonOptions);
|
|
52
121
|
run(options?: EvalComparisonRunOptions): Promise<ComparisonResult>;
|
|
53
122
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"eval-comparison.d.ts","sourceRoot":"","sources":["../src/eval-comparison.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,UAAU,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAChE,OAAO,KAAK,EAAE,QAAQ,EAAiC,mBAAmB,EAAE,MAAM,iBAAiB,CAAC;AACpG,OAAO,KAAK,
|
|
1
|
+
{"version":3,"file":"eval-comparison.d.ts","sourceRoot":"","sources":["../src/eval-comparison.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,UAAU,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAChE,OAAO,KAAK,EAAE,QAAQ,EAAiC,mBAAmB,EAAE,MAAM,iBAAiB,CAAC;AACpG,OAAO,KAAK,EAAwC,gBAAgB,EAAE,MAAM,UAAU,CAAC;AAEvF,OAAO,EAAE,OAAO,EAAE,MAAM,YAAY,CAAC;AAIrC,OAAO,KAAK,EAAE,eAAe,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAWlE,MAAM,WAAW,qBAAqB;IACpC,OAAO,EAAE,OAAO,CAAC;IACjB,OAAO,EAAE;QACP,QAAQ,EAAE,UAAU,CAAC;QACrB,UAAU,EAAE,UAAU,CAAC;KACxB,CAAC;IACF,OAAO,CAAC,EAAE,QAAQ,EAAE,CAAC;IACrB,kBAAkB,CAAC,EAAE,mBAAmB,EAAE,CAAC;IAC3C,gEAAgE;IAChE,KAAK,CAAC,EAAE,gBAAgB,CAAC;IACzB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB;;;OAGG;IACH,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB;;;OAGG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;;OAGG;IACH,IAAI,CAAC,EAAE,GAAG,GAAG,UAAU,CAAC;IACxB,4FAA4F;IAC5F,IAAI,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC;IACvB;;;OAGG;IACH,QAAQ,CAAC,EAAE,eAAe,CAAC;IAC3B,UAAU,CAAC,EAAE,CAAC,QAAQ,EAAE;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,KAAK,IAAI,CAAC;CACvF;AAED,MAAM,WAAW,wBAAwB;IACvC,kEAAkE;IAClE,MAAM,CAAC,EAAE,WAAW,CAAC;CACtB;AAED;;;;GAIG;AACH,MAAM,MAAM,cAAc,GAAG,SAAS,GAAG,eAAe,GAAG,GAAG,GAAG,UAAU,GAAG,MAAM,CAAC;AAErF,MAAM,MAAM,gBAAgB,GAAG,UAAU,GAAG,YAAY,GAAG,KAAK,CAAC;AAEjE,MAAM,WAAW,gBAAgB;IAC/B,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,sFAAsF;IACtF,CAAC,EAAE,MAAM,CAAC;IACV,gEAAgE;IAChE,UAAU,EAAE,MAAM,CAAC;IACnB,8EAA8E;IAC9E,EAAE,EAAE;QAAE,GAAG,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC;IAClC,4EAA4E;IAC5E,UAAU,EAAE,MAAM,CAAC;IACnB,IAAI,EAAE,cAAc,CAAC;IACrB,6BAA6B;IAC7B,MAAM,EAAE,MAAM,CAAC;IACf,mEAAmE;IACnE,cAAc,EAAE,MAAM,CAAC;IACvB,wCAAwC;IACxC,WAAW,EAAE,OAAO,CAAC;IACrB;;;OAGG;IACH,MAAM,EAAE,gBAAgB,CAAC;CAC1B;AAED;;;GAGG;AACH,MAAM,WAAW,kBAAkB;IACjC,+CAA+C;IAC/C,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,CAAC,EAAE,MAAM,CAAC;IACV,kFAAkF;IAClF,UAAU,EAAE,MAAM,CAAC;IACnB,IAAI,EAAE,cAAc,CAAC;IACrB,MAAM,EAAE,MAAM,CAAC;IACf,WAAW,EAAE,OAAO,CAAC;IACrB,gGAAgG;IAChG,MAAM,EAAE,gBAAgB,CAAC;CAC1B;AAED,MAAM,WAAW,gBAAgB;IAC/B,OAAO,EAAE;QACP,MAAM,EAAE,gBAAgB,CAAC;QACzB,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,gBAAgB,CAAC,CAAC;QAC1C,yEAAyE;QACzE,OAAO,EAAE,kBAAkB,CAAC;QAC5B,sFAAsF;QACtF,IAAI,CAAC,EAAE,kBAAkB,CAAC;KAC3B,CAAC;IACF,QAAQ,EAAE,eAAe,CAAC;IAC1B,UAAU,EAAE,eAAe,CAAC;IAC5B,gEAAgE;IAChE,QAAQ,CAAC,EAAE,cAAc,CAAC;CAC3B;AAuID,qBAAa,cAAc;IACzB,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAwB;IAC7C,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAuB;IAC9C,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAe;gBAElC,IAAI,EAAE,qBAAqB;IAiCjC,GAAG,CAAC,OAAO,GAAE,wBAA6B,GAAG,OAAO,CAAC,gBAAgB,CAAC;CAwG7E"}
|