warfbro commited on
Commit
fe9a41e
·
verified ·
1 Parent(s): a270284

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +90 -4
README.md CHANGED
@@ -1,7 +1,93 @@
1
  ---
2
- license: apache-2.0
3
  language:
4
  - zh
5
- base_model:
6
- - empero-ai/Qwythos-9B-Claude-Mythos-5-1M-GGUF
7
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
 
2
  language:
3
  - zh
4
+ license: apache-2.0
5
+ pipeline_tag: token-classification
6
+ tags:
7
+ - pytorch
8
+ - transformers
9
+ - named-entity-recognition
10
+ - token-classification
11
+ - ner
12
+ - ernie
13
+ - crf
14
+ - chinese-nlp
15
+ - person-name-extraction
16
+ - financial-documents
17
+ library_name: transformers
18
+ base_model: nghuyong/ernie-3.0-base-zh
19
+ ---
20
+
21
+ # 人名提取 — Human Name Extraction
22
+
23
+ 基于 ERNIE 3.0 + CRF 的中文金融批复人名提取工具。双模型集成架构,差异自动交 LLM 裁定。
24
+
25
+ ## 模型说明
26
+
27
+ - **底模**:ERNIE 3.0 Base (Chinese),118M 参数
28
+ - **架构**:ERNIE 编码器 → Linear(768→3) → CRF (BIO 标注)
29
+ - **任务**:从金融监管批复标题中提取人名(B-PER / I-PER)
30
+ - **训练数据**:2,530 条正例 + 1,000 条对抗负例(地名/公司名误识别)
31
+
32
+ ### 双模型
33
+
34
+ | 权重文件 | 说明 |
35
+ |----------|------|
36
+ | `finetune_best.pt` | 全参数微调(ERNIE 不冻结) |
37
+ | `frozen_best.pt` | 冻结 ERNIE + 对抗训练 |
38
+
39
+ ## 快速开始
40
+
41
+ ```bash
42
+ pip install torch>=2.4.0 "transformers==4.46.0" pytorch-crf openpyxl pandas "numpy<2"
43
+
44
+ # 单模型提取
45
+ python extract.py input.xlsx output.xlsx
46
+
47
+ # 双模型集成(差异交 LLM 裁定)
48
+ python ensemble_extract.py input.xlsx output.xlsx
49
+ ```
50
+
51
+ ### 输入格式
52
+
53
+ xlsx:A 列(标题),L 列(机构名等需在标题中删除的内容)
54
+
55
+ ### 输出
56
+
57
+ | 列 | 内容 |
58
+ |----|------|
59
+ | A列 | 原始标题 |
60
+ | B列 | 提取人名(`、`分隔) |
61
+ | C列 | 方法(`一致`/`LLM裁定`) |
62
+ | D列 | 姓名字数 |
63
+
64
+ ## 流水线
65
+
66
+ ```
67
+ 标题 → L列机构名剔除 → 模型推理 → 去等N人 → 英文括号扩展 → 存在性校验 → 输出
68
+ ```
69
+
70
+ ## LLM 配置(集成裁定用)
71
+
72
+ 项目根目录 `llm` 文件,OpenAI 兼容格式:
73
+
74
+ ```
75
+ api_key=sk-xxxxxxxx
76
+ api_url=https://api.deepseek.com/chat/completions
77
+ model=deepseek-chat
78
+ ```
79
+
80
+ ## 文件结构
81
+
82
+ ```
83
+ ├── config.py # 全局配置
84
+ ├── model.py # 模型定义 (ErnieCRF)
85
+ ├── extract.py # 单模型提取入口
86
+ ├── ensemble_extract.py # 双模型集成提取
87
+ ├── llm_resolver.py # LLM API 调用
88
+ ├── rule.py # 规则提取(备选)
89
+ ├── finetune_best.pt # 模型A 权重
90
+ ├── frozen_best.pt # 模型B 权重
91
+ ├── llm # LLM 配置模板
92
+ └── README.md # 本文件
93
+ ```