docs(README): 更新项目文档
This commit is contained in:
@@ -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
|
||||
```
|
||||
Reference in New Issue
Block a user