Skip to content

Add unit-circle embedding benchmark project - #1

Merged
erinepshovel-code merged 1 commit into
mainfrom
claude/unit-circle-embedding-benchmark-hvjp5
Apr 14, 2026
Merged

erinepshovel-code merged 1 commit into
mainfrom
claude/unit-circle-embedding-benchmark-hvjp5

Conversation

@erinepshovel-code

Copy link
Copy Markdown
Owner

Full Replit-ready benchmark for testing whether a unit-circle /
number-base coordinate remapping improves retrieval, clustering, and
neighborhood preservation on top of proven embedding models.

  • src/transforms.py: full A0–A5 ablation ladder (identity, PCA,
    random rotation, UC direct, UC rank, UC broken-phase control)
  • src/benchmark.py: retrieval (Recall@k, MRR, nDCG), clustering
    (silhouette, NMI, ARI), geometry (neighbor overlap, Spearman,
    trustworthiness/continuity)
  • src/embed.py: SentenceTransformer wrapper with task-prefix support
  • src/config.py: model presets (nomic, bge-m3) and ablation configs
  • src/main.py: full ablation runner with JSON output and success checks
  • src/metrics.py: table formatting, delta-vs-baseline, success criteria
  • data/: 20-doc toy corpus (4 topics) + 10 queries + qrels
  • notebooks/sanity_checks.ipynb: bar charts, neighbor overlap, UC geometry
  • requirements.txt, pyproject.toml, .replit

https://claude.ai/code/session_01EWFjWpMFhdAcdpST3syGPE

Full Replit-ready benchmark for testing whether a unit-circle /
number-base coordinate remapping improves retrieval, clustering, and
neighborhood preservation on top of proven embedding models.

- src/transforms.py: full A0–A5 ablation ladder (identity, PCA,
  random rotation, UC direct, UC rank, UC broken-phase control)
- src/benchmark.py: retrieval (Recall@k, MRR, nDCG), clustering
  (silhouette, NMI, ARI), geometry (neighbor overlap, Spearman,
  trustworthiness/continuity)
- src/embed.py: SentenceTransformer wrapper with task-prefix support
- src/config.py: model presets (nomic, bge-m3) and ablation configs
- src/main.py: full ablation runner with JSON output and success checks
- src/metrics.py: table formatting, delta-vs-baseline, success criteria
- data/: 20-doc toy corpus (4 topics) + 10 queries + qrels
- notebooks/sanity_checks.ipynb: bar charts, neighbor overlap, UC geometry
- requirements.txt, pyproject.toml, .replit

https://claude.ai/code/session_01EWFjWpMFhdAcdpST3syGPE
@erinepshovel-code
erinepshovel-code merged commit 16ecca4 into main Apr 14, 2026
1 check passed
@erinepshovel-code
erinepshovel-code deleted the claude/unit-circle-embedding-benchmark-hvjp5 branch April 14, 2026 20:07
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants