Compare commits

...

2 Commits

Author SHA1 Message Date
jiang 6debcbc1bd Merge remote-tracking branch 'origin/dev' into dev 2025-03-09 14:54:52 +08:00
jiang 95843aae79 重构模型训练 2025-03-09 14:50:42 +08:00
1 changed files with 2 additions and 2 deletions

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