@@ -1729,20 +1729,20 @@ from paddlenlp import Taskflow
1729
1729
```
1730
1730
>>> cls = Taskflow("zero_shot_text_classification", schema=["这是一条好评", "这是一条差评"])
1731
1731
>>> cls("房间干净明亮,非常不错")
1732
- [{'predictions': [{'label': '这是一条好评', 'score': 0.9695149765679986 }], 'text_a': '房间干净明亮,非常不错'}]
1732
+ [{'predictions': [{'label': '这是一条好评', 'score': 0.9072999699439914 }], 'text_a': '房间干净明亮,非常不错'}]
1733
1733
>>> cls("东西还可以,但是快递非常慢,下次不会再买这家了。")
1734
- [{'predictions': [{'label': '这是一条差评', 'score': 0.903727367612172 }], 'text_a': '东西还可以,但是快递非常慢,下次不会再买这家了。'}]
1734
+ [{'predictions': [{'label': '这是一条差评', 'score': 0.9282672873429476 }], 'text_a': '东西还可以,但是快递非常慢,下次不会再买这家了。'}]
1735
1735
```
1736
1736
1737
1737
#### 意图识别
1738
1738
1739
1739
```
1740
1740
>>> from paddlenlp import Taskflow
1741
- >>> schema = ["病情诊断", "治疗方案", "病因分析", "指标解读", "就医建议", "疾病表述", "后果表述", "注意事项", "功效作用", "医疗费用", "其他" ]
1741
+ >>> schema = ["病情诊断", "治疗方案", "病因分析", "指标解读", "就医建议", "疾病表述", "后果表述", "注意事项", "功效作用", "医疗费用"]
1742
1742
>>> cls("先天性厚甲症去哪里治")
1743
- [{'predictions': [{'label': '就医建议', 'score': 0.9628814210597645 }], 'text_a': '先天性厚甲症去哪里治'}]
1743
+ [{'predictions': [{'label': '就医建议', 'score': 0.5494891306403806 }], 'text_a': '先天性厚甲症去哪里治'}]
1744
1744
>>> cls("男性小腹疼痛是什么原因?")
1745
- [{'predictions': [{'label': '病因分析', 'score': 0.9925820373324141 }], 'text_a': '男性小腹疼痛是什么原因?'}]
1745
+ [{'predictions': [{'label': '病因分析', 'score': 0.5763229815300723 }], 'text_a': '男性小腹疼痛是什么原因?'}]
1746
1746
```
1747
1747
1748
1748
#### 语义相似度计算
@@ -1751,30 +1751,30 @@ from paddlenlp import Taskflow
1751
1751
>>> from paddlenlp import Taskflow
1752
1752
>>> cls = Taskflow("zero_shot_text_classification", schema=["不同", "相同"])
1753
1753
>>> cls([["怎么查看合同", "从哪里可以看到合同"]])
1754
- [{'predictions': [{'label': '相同', 'score': 0.9775065319076257 }], 'text_a': '怎么查看合同', 'text_b': '从哪里可以看到合同'}]
1754
+ [{'predictions': [{'label': '相同', 'score': 0.9951385264364382 }], 'text_a': '怎么查看合同', 'text_b': '从哪里可以看到合同'}]
1755
1755
>>> cls([["为什么一直没有电话来确认借款信息", "为何我还款了,今天却接到客服电话通知"]])
1756
- [{'predictions': [{'label': '不同', 'score': 0.9918983379165037 }], 'text_a': '为什么一直没有电话来确认借款信息', 'text_b': '为何我还款了,今天却接到客服电话通知'}]
1756
+ [{'predictions': [{'label': '不同', 'score': 0.9991497973466908 }], 'text_a': '为什么一直没有电话来确认借款信息', 'text_b': '为何我还款了,今天却接到客服电话通知'}]
1757
1757
```
1758
1758
1759
1759
#### 蕴含推理
1760
1760
1761
1761
```
1762
1762
>>> from paddlenlp import Taskflow
1763
- >>> cls = Taskflow("zero_shot_text_classification", schema=["中立 ", "蕴含", "矛盾"])
1763
+ >>> cls = Taskflow("zero_shot_text_classification", schema=["无关 ", "蕴含", "矛盾"])
1764
1764
>>> cls([["一个骑自行车的人正沿着一条城市街道朝一座有时钟的塔走去。", "骑自行车的人正朝钟楼走去。"]])
1765
- [{'predictions': [{'label': '蕴含', 'score': 0.9944843058584897 }], 'text_a': '一个骑自行车的人正沿着一条城市街道朝一座有时钟的塔走去。', 'text_b': '骑自行车的人正朝钟楼走去。'}]
1765
+ [{'predictions': [{'label': '蕴含', 'score': 0.9931122738524856 }], 'text_a': '一个骑自行车的人正沿着一条城市街道朝一座有时钟的塔走去。', 'text_b': '骑自行车的人正朝钟楼走去。'}]
1766
1766
>>> cls([["一个留着长发和胡须的怪人,在地铁里穿着一件颜色鲜艳的衬衫。", "这件衬衫是新的。"]])
1767
- [{'predictions': [{'label': '中立 ', 'score': 0.6659998351201399 }], 'text_a': '一个留着长发和胡须的怪人,在地铁里穿着一件颜色鲜艳的衬衫。', 'text_b': '这件衬衫是新的。'}]
1767
+ [{'predictions': [{'label': '无关 ', 'score': 0.997680189334587 }], 'text_a': '一个留着长发和胡须的怪人,在地铁里穿着一件颜色鲜艳的衬衫。', 'text_b': '这件衬衫是新的。'}]
1768
1768
>>> cls([["一个穿着绿色衬衫的妈妈和一个穿全黑衣服的男人在跳舞。", "两人都穿着白色裤子。"]])
1769
- [{'predictions': [{'label': '矛盾', 'score': 0.9270557883904931 }], 'text_a': '一个穿着绿色衬衫的妈妈和一个穿全黑衣服的男人在跳舞。', 'text_b': '两人都穿着白色裤子。'}]
1769
+ [{'predictions': [{'label': '矛盾', 'score': 0.9666946163628479 }], 'text_a': '一个穿着绿色衬衫的妈妈和一个穿全黑衣服的男人在跳舞。', 'text_b': '两人都穿着白色裤子。'}]
1770
1770
```
1771
1771
1772
1772
#### 可配置参数说明
1773
1773
1774
1774
* ` batch_size ` :批处理大小,请结合机器情况进行调整,默认为1。
1775
1775
* ` task_path ` :自定义任务路径,默认为None。
1776
1776
* ` schema ` :定义任务标签候选集合。
1777
- * ` model ` :选择任务使用的模型,默认为` utc-large ` , 支持` utc-xbase ` , ` utc-base ` , ` utc-medium ` , ` utc-micro ` , ` utc-mini ` , ` utc-nano ` , ` utc-pico ` 。
1777
+ * ` model ` :选择任务使用的模型,默认为` utc-base ` , 支持` utc-xbase ` , ` utc-base ` , ` utc-medium ` , ` utc-micro ` , ` utc-mini ` , ` utc-nano ` , ` utc-pico ` 。
1778
1778
* ` max_seq_len ` :最长输入长度,包括所有标签的长度,默认为512。
1779
1779
* ` pred_threshold ` :模型对标签预测的概率在0~1之间,返回结果去掉小于这个阈值的结果,默认为0.5。
1780
1780
* ` precision ` :选择模型精度,默认为` fp32 ` ,可选有` fp16 ` 和` fp32 ` 。` fp16 ` 推理速度更快。如果选择` fp16 ` ,请先确保机器正确安装NVIDIA相关驱动和基础软件,** 确保CUDA>=11.2,cuDNN>=8.1.1** ,初次使用需按照提示安装相关依赖。其次,需要确保GPU设备的CUDA计算能力(CUDA Compute Capability)大于7.0,典型的设备包括V100、T4、A10、A100、GTX 20系列和30系列显卡等。更多关于CUDA Compute Capability和精度支持情况请参考NVIDIA文档:[ GPU硬件与支持精度对照表] ( https://docs.nvidia.com/deeplearning/tensorrt/archives/tensorrt-840-ea/support-matrix/index.html#hardware-precision-matrix ) 。
0 commit comments