mobile wallpaper 1mobile wallpaper 2mobile wallpaper 3mobile wallpaper 4
402 字
1 分钟
本地 Embedding 模型怎么选:别从参数量开始,先做检索小测
2025-05-22

本地 Embedding 模型怎么选:别从参数量开始,先做检索小测#

做本地代码库索引时,我不会先按参数量选 Embedding 模型。硬件、语料语言、查询方式和向量库都会影响效果,任何模型推荐都应先在目标语料上做小规模验证。

Embedding 的作用是把文本转换成向量,供相似内容检索;模型大不等于你的检索一定更准。选择前先准备几十条真实查询和对应的正确文件或段落,建立一份小型测试集。

一小时内能做的对比#

  1. 固定相同的文档切块、向量数据库和查询集;
  2. 对每个候选模型记录 Recall@k:正确材料是否出现在前 k 个结果里;
  3. 测量索引耗时、查询延迟、显存 / 内存和磁盘占用;
  4. 看失败案例:代码符号、中文技术词、跨语言名称是否被正确召回;
  5. 再决定是否值得用更大的模型或托管 API。

代码库检索尤其需要把符号名、路径、README 和代码片段混在测试集中。部署阶段还要确认查询端和索引端使用同一个模型及相同的归一化方式,否则向量空间不一致,召回会悄悄变差。

资料来源:本地跑 Embedding 模型推荐(讨论提供候选方向,没有统一性能测试)

分享

如果这篇文章对你有帮助,欢迎分享给更多人!

本地 Embedding 模型怎么选:别从参数量开始,先做检索小测
https://blog.zpooi.com/posts/local-embedding-model-selection/
作者
zpooi
发布于
2025-05-22
许可协议
CC BY-NC-SA 4.0

部分信息可能已经过时

目录