ShareGPT数据集中的代码对话提取:筛选编程相关对话用于代码模型训练的方法

夜萱吖_2822

夜萱吖_2822

2026-05-24

278人浏览

原创

需从sharegpt中精准提取编程相关对话,方法包括:一、关键词与正则初步过滤;二、tree-sitter语法解析验证;三、distilbert微调分类器判别语义意图;四、角色与上下文行为连贯性分析;五、许可证合规筛选及哈希去重。

☞☞☞AI 智能聊天, 问答助手, AI 智能搜索, 多模态理解力帮你轻松跨越从0到1的创作门槛☜☜☜

sharegpt数据集中的代码对话提取:筛选编程相关对话用于代码模型训练的方法

如果您在使用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对话样本,标签为“编程相关”或“非编程相关”,重点覆盖指令模糊、无代码输出、跨语言混用等边界案例。

ARC Lab
ARC Lab

一款提供AI人像处理、图像增强和智能抠图等能力的视觉处理工具集合,方便用户在线体验多种图片优化功能。

下载

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),合并相同内容的不同对话实例,仅保留最早出现的一条,防止数据偏斜。

相关专题

更多
OpenJDK JVM参数怎么调优
OpenJDK JVM参数怎么调优

Java应用运行缓慢,如何调整JVM参数?本专题介绍OpenJDK JVM性能调优的基本流程,涵盖性能指标确认、运行数据采集、内存配置和垃圾回收分析,帮助你先定位瓶颈,再通过对照测试验证效果,避免直接套用参数而影响应用稳定性。

2026.10.16

0

10

OpenJDK怎么配置Java开发环境
OpenJDK怎么配置Java开发环境

怎样用OpenJDK搭建开发环境?本专题介绍版本选择、下载安装、环境变量配置和编译运行验证,帮助你理清JDK的作用及基本配置流程,并通过简单程序检查环境是否可用,为后续项目开发做好准备。

2026.10.16

0

10

OpenJDK怎么部署Java应用
OpenJDK怎么部署Java应用

Java项目开发完成后,怎样部署到目标环境并正常运行?本专题介绍OpenJDK应用部署流程,涵盖运行环境准备、版本兼容检查、应用打包、启动参数配置和日志查看,帮助你理清部署前后的操作顺序,减少环境差异造成的运行问题。

2026.10.16

0

10

Revel框架模板渲染与表单验证
Revel框架模板渲染与表单验证

本专题围绕Revel框架视图层开发,介绍模板语法、变量输出、布局复用、表单提交、参数绑定与验证规则,帮助开发者完成安全规范的页面交互。

2026.10.16

0

15

Revel框架MVC路由与控制器
Revel框架MVC路由与控制器

本专题聚焦Revel框架MVC核心机制,讲解路由定义、控制器Action、参数绑定、请求响应和反向路由,并结合常见场景梳理接口请求处理流程。

2026.10.16

0

15

Revel框架入门与安装配置
Revel框架入门与安装配置

本专题从Revel框架环境搭建讲起,系统介绍命令行工具安装、项目创建、目录结构、配置文件及热代码重载,帮助开发者快速完成首个Go Web应用。

2026.10.16

0

15

Linux 新手先学哪些命令
Linux 新手先学哪些命令

Linux新手命令专题汇集终端基础、文件目录操作、文本查看、软件安装和权限管理等入门文章,帮助按实际任务练习常用命令并建立命令行操作习惯。

2026.10.15

20

20

MySQL 忘记密码怎么办
MySQL 忘记密码怎么办

MySQL忘记密码专题汇集不同版本与系统下的root密码重置方法,覆盖MySQL 5.7、8.0、Mac、Linux及本地集成环境,并提示恢复权限与重新登录的检查步骤。

2026.10.15

20

18

Vue.js 中 ref 和 reactive 的区别
Vue.js 中 ref 和 reactive 的区别

Vue.js中ref和reactive的区别专题汇集两种响应式API的定义、访问方式、对象解构、侦听与性能取舍等文章,帮助结合实际状态类型选择合适写法。

2026.10.15

0

20

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
热门推荐
/
最新课程
phpStudy极速入门视频教程
phpStudy极速入门视频教程

共6课时 | 54.7万人学习

独孤九贱(4)_PHP视频教程
独孤九贱(4)_PHP视频教程

共89课时 | 133.7万人学习