Skip to content

Commit

Permalink
Add plato-mini for taskflow (#1383)
Browse files Browse the repository at this point in the history
* 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
  • Loading branch information
linjieccc authored Dec 17, 2021
1 parent 1ecff72 commit 5f2862a
Show file tree
Hide file tree
Showing 4 changed files with 423 additions and 5 deletions.
46 changes: 44 additions & 2 deletions docs/model_zoo/taskflow.md
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,7 @@
- [知识挖掘-名词短语标注](#知识挖掘-名词短语标注)
- [生成式问答](#生成式问答)
- [智能写诗](#智能写诗)
- [开放域对话](#开放域对话)
- [FAQ](#FAQ)

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

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

### 开放域对话

非交互模式:
```python
from paddlenlp import Taskflow

dialogue = Taskflow("dialogue")
dialogue(["吃饭了吗"])
>>> ['刚吃完饭,你在干什么呢?']

dialogue(["你好", "吃饭了吗"], ["你是谁?"])
>>> ['吃过了,你呢', '我是李明啊']
```

可配置参数:

* `batch_size`:批处理大小,请结合机器情况进行调整,默认为1。
* `max_seq_len`:最大序列长度,默认为512。

交互模式:
```python
from paddlenlp import Taskflow

dialogue = Taskflow("dialogue")
# 输入`exit`可退出交互模式
dialogue.interactive_mode(max_turn=3)

'''
[Human]:你好
[Bot]:你好,很高兴认识你,我想问你一下,你喜欢运动吗?
[Human]:喜欢
[Bot]:那你喜欢什么运动啊?
[Human]:篮球,你喜欢篮球吗
[Bot]:当然了,我很喜欢打篮球的?
'''
```

交互模式参数:

* `max_turn`:任务能记忆的对话轮数,当max_turn为1时,模型只能记住当前对话,无法获知之前的对话内容。

## FAQ

### Q1 Taskflow如何修改任务保存路径?
Expand Down
Loading

0 comments on commit 5f2862a

Please sign in to comment.