Text Classification
Transformers
ONNX
Safetensors
Chinese
bert
chinese
intent-classification
text-embeddings-inference
Instructions to use pawizard/traffic-classify with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use pawizard/traffic-classify with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="pawizard/traffic-classify")# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("pawizard/traffic-classify") model = AutoModelForSequenceClassification.from_pretrained("pawizard/traffic-classify", device_map="auto") - Notebooks
- Google Colab
- Kaggle
File size: 3,955 Bytes
aa7739c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 | # 数据源说明
本项目当前使用三分类标签:
| label | 含义 |
| --- | --- |
| 0 | 非研发相关 |
| 1 | 研发相关 |
| 2 | 中性 |
## 当前推荐数据集
推荐使用精简后的数据集:
- 原始精简 CSV:`data/refined_compact_labeled.csv`
- 训练集:`data_refined/train.csv`
- 验证集:`data_refined/val.csv`
- 测试集:`data_refined/test.csv`
当前精简 CSV 的类别分布:
| label | 含义 | 数量 |
| --- | --- | ---: |
| 0 | 非研发相关 | 22337 |
| 1 | 研发相关 | 68779 |
| 2 | 中性 | 2312 |
切分比例由 `data_prep.py` 生成,约为 `8:1:1`。
## 数据来源
### THUOCL
- 仓库:https://github.com/thunlp/THUOCL
- 用途:提供中文领域词库。
- 中间产物:`data/thuocl_labeled.csv`
初始映射中,`THUOCL_IT.txt` 标为研发相关,其他领域曾用于非研发或中性候选。当前精简数据集中仅保留部分高权重词:
- `IT`:研发相关,全量保留去重后的词条
- 其他 THUOCL 领域:非研发相关,每个领域最多保留 300 条
### DomainWordsDict
- 仓库:https://github.com/liuhuanyong/DomainWordsDict
- 用途:提供大规模中文领域词库。
- 中间产物:`data/domain_words_labeled.csv`
当前精简映射:
- `计算机业`:研发相关,全量保留去重后的词条
- `电子工程`:研发相关,全量保留去重后的词条
- `通信工程`:研发相关,全量保留去重后的词条
- 其他领域:非研发相关,每个领域最多保留 300 条
旧版曾把 `网络用语`、`诗词歌赋`、`文学名著`、`汉语言学` 标为中性;当前已取消这个策略,这些领域在精简数据集中都按非研发相关处理。
### computerese-cross-references
- 仓库:https://github.com/EarsEyesMouth/computerese-cross-references
- 用途:补充计算机专业术语中英文对照。
- 中间产物:`data/computerese_labeled.csv`
该仓库主要内容在 `README.md` 的 Markdown 表格中。当前抽取策略:
- 英文术语全部标为研发相关
- 中文释义只保留明显具有技术语义的短语
- 过滤掉过于泛化的中文词,例如“获取”这类容易导致误判的词
当前精简数据集中,`computerese` 全量保留去重后的词条,约 1650 条。
### stopwords
- 仓库:https://github.com/goto456/stopwords
- 用途:提供中性类样本。
- 使用文件:
- `baidu_stopwords.txt`
- `cn_stopwords.txt`
- `hit_stopwords.txt`
- `scu_stopwords.txt`
当前策略:
- 所有去重后的停用词标为 `2=中性`
- 中性类只来自该停用词仓库
- 例如“继续”会被标为中性
## 生成策略
精简数据集的目标是避免中性和非研发样本过多,降低类别不均衡和噪声。
当前规则:
| 类别 | 来源 | 保留策略 |
| --- | --- | --- |
| 研发相关 | `IT`、`计算机业`、`电子工程`、`通信工程`、`computerese` | 全量保留去重后的词条,并优先于非研发来源 |
| 中性 | stopwords | 全量去重保留 |
| 非研发相关 | 其他领域词库 | 每个领域最多 300 条 |
生成后的 CSV 字段:
| 字段 | 含义 |
| --- | --- |
| `text` | 训练文本 |
| `label` | 标签 ID |
| `label_name` | 标签名称 |
| `category` | 来源领域或类别 |
| `df` | 原词库权重或词频,部分来源为空 |
| `source_file` | 来源文件 |
## 注意事项
- 当前数据主要是词库,不是完整用户 query。训练出的模型更偏向领域词识别,后续应补充真实用户话术。
- 中性类目前使用停用词近似表达,如“继续”“好的”“然后”等,适合覆盖短指令和弱语义输入。
- 非研发相关按领域抽样压缩后,与研发相关数量大致均衡,适合先训练一个轻量 baseline。
- 若后续新增真实标注样本,建议优先加入 `data/refined_compact_labeled.csv` 或单独维护一个人工标注 CSV,再合并后重新切分。
|