docs(README): 更新项目文档

This commit is contained in:
2025-08-01 21:04:29 +08:00
parent 092921f401
commit 5a1dc4ac38
+49 -1
View File
@@ -1,3 +1,51 @@
## KHAOSZ-dataset
用于训练的KHAOSZ数据集
用于训练的KHAOSZ数据集
### 项目结构
``` bash
.
├── pre_train/ # 预训练数据处理脚本
│ ├── chinese-c4.py # 中文预训练数据处理
│ ├── chinese-cosmopedia.py # 中文维基数据处理
│ ├── english-fineweb.py # 英文网页数据处理
│ └── english-wiki.py # 英文维基数据处理
├── supervised_finetuning/ # 监督微调数据处理脚本
│ ├── sft_belle.py # BelleGroup对话数据处理
│ ├── sft_chinese_instruct.py # 中文指令微调数据处理
│ ├── sft_coder.py # 编程对话数据处理
│ └── sft_magpie-pro-300k.py # Magpie对话数据处理
├── modules/
│ ├── utils.py # 核心数据处理工具(规范化/打包/序列化)
│ └── tokenizer.py # BPE分词器实现(支持自定义特殊token)
├── dump_file/
│ ├── dump_pt_file.py # 预训练数据二进制打包
│ └── dump_sft_file.py # 微调数据二进制打包
├── run.py # 主执行脚本(支持subprocess调用)
└── README.md # 项目文档
```
### 数据集特性
- 支持多语言混合训练(中/英文)
- 包含以下预训练数据源:
- Chinese-C4
- Chinese-Cosmopedia
- English-Fineweb
- English-Wiki
- 支持监督微调数据集:
- Ling-Coder-SFT
- Chinese-Instruct
- BelleGroup
- Magpie-Pro-300K
### 使用说明
1. 安装依赖:
```bash
pip install datasets tokenizers tqdm torch
```
运行数据处理:
```bash
python run.py
```