-
Notifications
You must be signed in to change notification settings - Fork 2.9k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add UIE-M #3192
Add UIE-M #3192
Conversation
@@ -886,6 +888,41 @@ from paddlenlp import Taskflow | |||
[{'时间': [{'text': '2月8日上午', 'start': 0, 'end': 6, 'probability': 0.6513581678349247}], '选手': [{'text': '谷爱凌', 'start': 28, 'end': 31, 'probability': 0.9819330659468051}], '赛事名称': [{'text': '北京冬奥会自由式滑雪女子大跳台决赛', 'start': 6, 'end': 23, 'probability': 0.4908131110420939}]}] | |||
``` | |||
|
|||
- `uie-m-base`和`uie-m-large`支持中英文混合抽取,调用示例: |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
这里确定一下UIE多语言版本,在训练的时候只是支持中文、英文
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
嗯嗯 目前UIE-M多任务阶段只加了中英文数据
model_zoo/uie/README.md
Outdated
batch_size=1, | ||
model='uie-base', | ||
position_prob=0.5, | ||
precision='fp32') | ||
``` | ||
|
||
* `schema`:定义任务抽取目标,可参考开箱即用中不同任务的调用示例进行配置。 | ||
* `schema_lang`:设置schema的语言,因为中英schema的构造有所不同,因此需要指定schema的语言。该参数只对`uie-m-base`和`uie-m-large`模型有效。 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
这里应该提示设置哪些内容
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
done, thx
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
PR types
New features
PR changes
Models
Description
新增
uie-m-base
和uie-m-large
支持中英文混合抽取,调用示例:模型结构:
uie-m-large
uie-m-base
模型指标: