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

2
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. 进行预测