Datatrove v0.5.0 版本发布:数据处理工具链的重大升级
2025-06-19 02:45:36作者:翟萌耘Ralph
Datatrove 是一个专注于大规模文本数据处理的开源工具库,它为数据清洗、预处理和转换提供了高效且灵活的解决方案。该项目特别适用于自然语言处理(NLP)领域的数据准备工作,能够帮助研究人员和工程师处理海量文本数据。
核心功能改进
1. 分词器功能增强
新版本对分词器功能进行了多项改进:
- 增加了对本地Hugging Face分词器文件夹的直接加载支持,简化了本地化部署流程
- 引入了Megatron分词管道,为大规模模型训练提供了更好的支持
- 文档分词器现在支持基于块(chunk)的随机打乱功能,提高了数据处理的灵活性
2. 数据去重优化
URL去重功能得到了显著改进:
- 修复了使用索引时URL去重的问题,提高了去重准确性
- 为哈希索引添加了glob模式支持,使得大规模数据集处理更加高效
- MinhashBuildIndex中的签名优先级队列初始化问题得到修复
3. 数据处理流程增强
- 新增了RayPipelineExecutor,为分布式处理提供了新的执行引擎选项
- 允许自定义Parquet模式,提高了数据存储的灵活性
- 改进了元数据解析功能,特别是针对finemath场景
系统优化与稳定性提升
1. 资源管理改进
- 增加了OOM(内存不足)评分机制,帮助更好地监控和管理内存使用
- 缩短了轮询间隔,提高了系统响应速度
- 增强了依赖检查机制,现在会包括pip发行版的检查
2. SLURM集成增强
- 添加了
job_id_position参数到launch_slurm_job方法 - 支持自定义获取SLURM作业ID的方式
- 这些改进使得在高性能计算集群上的作业调度更加灵活
文本处理质量提升
- 调整了FTFY(修复文本)的默认设置,提高了文本修复的质量
- 修复了停用词实现中的问题,使文本过滤更加准确
- 改进了元数据解析,特别是针对数学相关内容
性能优化
- 哈希索引现在支持glob模式,提高了大规模数据处理的效率
- 文档分词器合并器现在支持基于块的随机打乱,优化了内存使用
- 当设置return_positions=True时,现在会基于.index返回位置信息
总结
Datatrove v0.5.0版本带来了全面的功能增强和性能优化,特别是在分词处理、数据去重和分布式执行方面有了显著进步。这些改进使得该工具库在大规模文本数据处理场景下更加可靠和高效。对于从事NLP研究和应用开发的团队来说,这个版本提供了更多灵活性和更好的性能表现,值得升级使用。
登录后查看全文
热门项目推荐
AutoGLM-Phone-9BAutoGLM-Phone-9B是基于AutoGLM构建的移动智能助手框架,依托多模态感知理解手机屏幕并执行自动化操作。Jinja00
Kimi-K2-ThinkingKimi K2 Thinking 是最新、性能最强的开源思维模型。从 Kimi K2 开始,我们将其打造为能够逐步推理并动态调用工具的思维智能体。通过显著提升多步推理深度,并在 200–300 次连续调用中保持稳定的工具使用能力,它在 Humanity's Last Exam (HLE)、BrowseComp 等基准测试中树立了新的技术标杆。同时,K2 Thinking 是原生 INT4 量化模型,具备 256k 上下文窗口,实现了推理延迟和 GPU 内存占用的无损降低。Python00
GLM-4.6V-FP8GLM-4.6V-FP8是GLM-V系列开源模型,支持128K上下文窗口,融合原生多模态函数调用能力,实现从视觉感知到执行的闭环。具备文档理解、图文生成、前端重构等功能,适用于云集群与本地部署,在同类参数规模中视觉理解性能领先。Jinja00
HunyuanOCRHunyuanOCR 是基于混元原生多模态架构打造的领先端到端 OCR 专家级视觉语言模型。它采用仅 10 亿参数的轻量化设计,在业界多项基准测试中取得了当前最佳性能。该模型不仅精通复杂多语言文档解析,还在文本检测与识别、开放域信息抽取、视频字幕提取及图片翻译等实际应用场景中表现卓越。00
GLM-ASR-Nano-2512GLM-ASR-Nano-2512 是一款稳健的开源语音识别模型,参数规模为 15 亿。该模型专为应对真实场景的复杂性而设计,在保持紧凑体量的同时,多项基准测试表现优于 OpenAI Whisper V3。Python00
GLM-TTSGLM-TTS 是一款基于大语言模型的高质量文本转语音(TTS)合成系统,支持零样本语音克隆和流式推理。该系统采用两阶段架构,结合了用于语音 token 生成的大语言模型(LLM)和用于波形合成的流匹配(Flow Matching)模型。 通过引入多奖励强化学习框架,GLM-TTS 显著提升了合成语音的表现力,相比传统 TTS 系统实现了更自然的情感控制。Python00
Spark-Formalizer-X1-7BSpark-Formalizer 是由科大讯飞团队开发的专用大型语言模型,专注于数学自动形式化任务。该模型擅长将自然语言数学问题转化为精确的 Lean4 形式化语句,在形式化语句生成方面达到了业界领先水平。Python00
项目优选
收起
deepin linux kernel
C
24
9
Ascend Extension for PyTorch
Python
223
245
Nop Platform 2.0是基于可逆计算理论实现的采用面向语言编程范式的新一代低代码开发平台,包含基于全新原理从零开始研发的GraphQL引擎、ORM引擎、工作流引擎、报表引擎、规则引擎、批处理引引擎等完整设计。nop-entropy是它的后端部分,采用java语言实现,可选择集成Spring框架或者Quarkus框架。中小企业可以免费商用
Java
9
1
暂无简介
Dart
672
157
本项目是CANN提供的数学类基础计算算子库,实现网络在NPU上加速计算。
C++
662
313
React Native鸿蒙化仓库
JavaScript
262
323
🔥LeetCode solutions in any programming language | 多种编程语言实现 LeetCode、《剑指 Offer(第 2 版)》、《程序员面试金典(第 6 版)》题解
Java
64
19
仓颉编译器源码及 cjdb 调试工具。
C++
134
867
仓颉编程语言测试用例。
Cangjie
37
860
openGauss kernel ~ openGauss is an open source relational database management system
C++
160
218