Paddle 一键预测

https://github.com/PaddlePaddle/PaddleNLP

PaddleNLP提供一键预测功能,无需训练,直接输入数据即可开放域抽取结果:

from pprint import pprint
from paddlenlp import Taskflow
schema = ['时间', '选手', '赛事名称']


ie = Taskflow('information_extraction', schema=schema)
txt ='2月8日上午北京冬奥会自由式滑雪女子大跳台决赛中中国选手谷爱凌以188.25分获得金牌!'
txt=ie(txt)
pprint(txt)

输出

[{'时间': [{'end': 6,
          'probability': 0.9857378532473966,
          'start': 0,
          'text': '2月8日上午'}],
  '赛事名称': [{'end': 23,
            'probability': 0.8503081103003893,
            'start': 6,
            'text': '北京冬奥会自由式滑雪女子大跳台决赛'}],
  '选手': [{'end': 31,
          'probability': 0.8981535684051067,
          'start': 28,
          'text': '谷爱凌'}]}]

 

posted on 2023-02-13 17:08  shaomine  阅读(61)  评论(0编辑  收藏  举报