需从sharegpt中精准提取编程相关对话,方法包括:一、关键词与正则初步过滤;二、tree-sitter语法解析验证;三、distilbert微调分类器判别语义意图;四、角色与上下文行为连贯性分析;五、许可证合规筛选及哈希去重。
☞☞☞AI 智能聊天, 问答助手, AI 智能搜索, 多模态理解力帮你轻松跨越从0到1的创作门槛☜☜☜

如果您在使用ShareGPT数据集训练代码模型时发现大量对话与编程无关,导致模型学习噪声增多、泛化能力下降,则需从原始对话中精准提取编程相关子集。以下是筛选编程相关对话的具体方法:
一、基于关键词与正则模式的初步过滤
该方法通过预定义的编程语言标识符、技术术语及代码结构特征进行快速粗筛,适用于大规模语料的首轮去噪,能高效剔除明显非编程类对话(如日常闲聊、情感咨询等)。
1、构建编程强相关关键词列表,包括语言名(python、java、rust、typescript)、框架名(React、Spring Boot、PyTorch)、工具链(git、Docker、pip、npm)及高频动词(debug、compile、refactor、serialize)。
2、编写正则表达式匹配典型代码片段结构,例如以```开头结尾的代码块、缩进4空格/Tab的多行逻辑块、含def/class/fn/func关键字的声明行、含import/require/use的依赖引入行。
3、对每条ShareGPT对话中的user与assistant消息分别执行关键词命中+正则匹配双条件判断,仅当至少一条消息同时满足“≥2个关键词命中”且“存在1处有效代码结构匹配”时,整轮对话被保留。
二、基于代码语法解析器的结构验证
该方法利用真实编程语言的语法解析器(如tree-sitter)对候选对话中提取出的代码块进行AST级校验,排除伪代码、拼写错误或语法无效的文本,确保保留的代码片段具备可编译/可解释基础。
1、从已通过关键词过滤的对话中,用正则提取所有```lang...```包裹块及无围栏但符合缩进/关键字特征的连续代码段。
2、为每种目标语言加载对应tree-sitter语言绑定(如tree-sitter-python、tree-sitter-javascript),对每个代码段调用parse()函数。
3、若解析返回有效AST根节点且错误节点数为0,则标记该代码段为语法合法;单条对话中任一代码段通过验证,即视为编程相关对话。
三、基于微调分类器的语义判别
该方法使用轻量级文本分类模型(如DistilBERT-finetuned-on-code-dialogue)对整轮对话做二分类,判断其是否属于“编程意图对话”,弥补关键词与语法方法在隐含意图(如“帮我写个爬虫”未出现代码块但明确指向编程任务)上的识别盲区。
1、准备标注数据:人工标注2000条ShareGPT对话样本,标签为“编程相关”或“非编程相关”,重点覆盖指令模糊、无代码输出、跨语言混用等边界案例。
2、在标注集上微调DistilBERT-base-uncased,输入为user消息+assistant消息拼接,最大长度512,训练3轮,验证集F1达0.92。
3、将微调后模型部署为本地推理服务,对全量ShareGPT对话逐条打分;仅当预测概率≥0.85时,判定为编程相关对话并纳入训练集。
四、基于对话角色与上下文连贯性的行为分析
该方法考察对话中用户提问与助手响应之间的技术一致性,例如用户请求修复Bug,助手是否提供可运行修复方案并附带错误定位;避免保留“用户问编程,助手答鸡汤”类低质量配对。
1、定义四类编程对话行为模式:问题诊断(含error trace)、代码生成(含完整函数/类)、调试指导(含step-by-step执行建议)、文档解释(含API签名与参数说明)。
2、对每轮对话的assistant响应,使用规则+小模型联合识别其是否归属上述任一类;同时检查user消息是否包含对应触发信号(如“报错”“怎么写”“解释下”“为什么”)。
3、仅当user触发信号与assistant响应行为类型匹配,且响应中包含至少1个可验证技术要素(如具体函数名、行号、异常类名、参数键名),才确认该对话具备编程行为连贯性。
五、基于许可证与代码来源可信度的合规筛选
该方法聚焦于保障提取出的编程对话中所含代码片段的再分发合法性,规避GPL等强传染性协议污染,确保下游代码模型训练符合开源合规要求。
1、对所有保留对话中的代码块,调用license-expression库解析其头部注释或关联README中的许可证声明,支持SPDX标准格式(如MIT、Apache-2.0、BSD-3-Clause)。
2、排除含GPL-2.0、GPL-3.0、AGPL等Copyleft许可证标识的代码块;若代码块无显式声明,则反向检索其GitHub仓库URL(若对话中提供),抓取对应LICENSE文件进行二次验证。
3、对通过许可证校验的代码块,执行哈希去重(SHA256),合并相同内容的不同对话实例,仅保留最早出现的一条,防止数据偏斜。











