weiweiw
/
Intention
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

重构模型训练

This commit is contained in:
jiang 2025-03-09 14:50:42 +08:00
parent aabc01d209
commit 95843aae79
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
uie/test_model.py
View File

@ -7,7 +7,7 @@ model = ErnieForTokenClassification.from_pretrained(model_path)
tokenizer = ErnieTokenizer.from_pretrained(model_path) tokenizer = ErnieTokenizer.from_pretrained(model_path)
# 2. 处理输入文本 # 2. 处理输入文本
text = "能否帮我进入日计划应用" text = "今天杨柳220kV变电站220kV南坪间隔扩建工程有多少作业计划"
inputs = tokenizer(text, max_len=512, return_tensors="pd") inputs = tokenizer(text, max_len=512, return_tensors="pd")
# 3. 进行预测 # 3. 进行预测
@ -29,7 +29,7 @@ label_map = {
8: 'B-subcontractor', 19: 'I-subcontractor', 8: 'B-subcontractor', 19: 'I-subcontractor',
9: 'B-team_leader', 20: 'I-team_leader', 9: 'B-team_leader', 20: 'I-team_leader',
10: 'B-risk_level', 21: 'I-risk_level', 10: 'B-risk_level', 21: 'I-risk_level',
11:'B-page',22:'I-page', 11: 'B-page', 22: 'I-page',
} }
# 5. 解析预测结果 # 5. 解析预测结果