Skip to content

Commit 5f2862a

Browse files
authored
Add plato-mini for taskflow (#1383)
* Add plato-mini for taskflow * Update README.md * Update README.md * Update README.md * Update model generation * Update task display * Update interactive mode * Update usage
1 parent 1ecff72 commit 5f2862a

File tree

4 files changed

+423
-5
lines changed

4 files changed

+423
-5
lines changed

docs/model_zoo/taskflow.md

Lines changed: 44 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
- [知识挖掘-名词短语标注](#知识挖掘-名词短语标注)
1717
- [生成式问答](#生成式问答)
1818
- [智能写诗](#智能写诗)
19+
- [开放域对话](#开放域对话)
1920
- [FAQ](#FAQ)
2021

2122
## 介绍
@@ -28,8 +29,8 @@
2829
| :------------ | ---- |
2930
| 中文分词 | 生成式问答 |
3031
| 词性标注 | 智能写诗 |
31-
| 命名实体识别 | 文本翻译(TODO) |
32-
| 文本纠错 | 开放域对话(TODO) |
32+
| 命名实体识别 | 开放域对话 |
33+
| 文本纠错 | 文本翻译(TODO) |
3334
| 句法分析 | 自动对联(TODO) |
3435
| 情感分析 | |
3536
| 文本相似度 | |
@@ -452,6 +453,47 @@ poetry(["林密不见人", "举头邀明月"])
452453

453454
* `batch_size`:批处理大小,请结合机器情况进行调整,默认为1。
454455

456+
### 开放域对话
457+
458+
非交互模式:
459+
```python
460+
from paddlenlp import Taskflow
461+
462+
dialogue = Taskflow("dialogue")
463+
dialogue(["吃饭了吗"])
464+
>>> ['刚吃完饭,你在干什么呢?']
465+
466+
dialogue(["你好", "吃饭了吗"], ["你是谁?"])
467+
>>> ['吃过了,你呢', '我是李明啊']
468+
```
469+
470+
可配置参数:
471+
472+
* `batch_size`:批处理大小,请结合机器情况进行调整,默认为1。
473+
* `max_seq_len`:最大序列长度,默认为512。
474+
475+
交互模式:
476+
```python
477+
from paddlenlp import Taskflow
478+
479+
dialogue = Taskflow("dialogue")
480+
# 输入`exit`可退出交互模式
481+
dialogue.interactive_mode(max_turn=3)
482+
483+
'''
484+
[Human]:你好
485+
[Bot]:你好,很高兴认识你,我想问你一下,你喜欢运动吗?
486+
[Human]:喜欢
487+
[Bot]:那你喜欢什么运动啊?
488+
[Human]:篮球,你喜欢篮球吗
489+
[Bot]:当然了,我很喜欢打篮球的?
490+
'''
491+
```
492+
493+
交互模式参数:
494+
495+
* `max_turn`:任务能记忆的对话轮数,当max_turn为1时,模型只能记住当前对话,无法获知之前的对话内容。
496+
455497
## FAQ
456498

457499
### Q1 Taskflow如何修改任务保存路径?

0 commit comments

Comments
 (0)