Python 实现语句中提取人名

7,038次阅读
没有评论

环境依赖

需要安装两个库,其实一个也可以,但是我这边准备了两个库做个比较。

安装命令如下:

pip install LAC -i https://pypi.douban.com/simple pip install ltp -i https://pypi.douban.com/simple

代码

  1. #!/user/bin/env python
  2. # coding=utf-8
  3. “””
  4. “””
  5. from LAC import LAC
  6. from ltp import LTP
  7. import time
  8. lac = LAC(mode=”lac”)
  9. ltp = LTP()
  10. # 句子提取名字
  11. def extract_name(sentence: strtype=’lac’):
  12.     user_name_lis = []
  13.     if type == ‘lac’:
  14.         _result = lac.run(sentence)
  15.         for _index, _label in enumerate(_result[1]):
  16.             if _label == “PER”:
  17.                 user_name_lis.append(_result[0][_index])
  18.     elif type == ‘ltp’:
  19.         _seg, _hidden = ltp.seg([sentence])
  20.         _pos_hidden = ltp.pos(_hidden)
  21.         for _seg_i, _seg_v in enumerate(_seg):
  22.             _hidden_v = _pos_hidden[_seg_i]
  23.             for _h_i, _h_v in enumerate(_hidden_v):
  24.                 if _h_v == “nh”:
  25.                     user_name_lis.append(_seg_v[_h_i])
  26.     else:
  27.         raise Exception(‘type not suppose’)
  28.     return user_name_lis
  29. if __name__ == ‘__main__‘:
  30.     _start_lac = time.time()
  31.     lis1 = extract_name(“就因为看了沈腾和贾玲的王牌对王牌节目,所以杨迪肯定偷题了。”, ‘lac’)
  32.     _end_lac = time.time()
  33.     print(“LAC: {} <- 耗时[{}]秒”.format(lis1, (_end_lac – _start_lac)))
  34.     _start_ltp = time.time()
  35.     lis2 = extract_name(“就因为看了沈腾和贾玲的王牌对王牌节目,所以杨迪肯定偷题了。”, ‘ltp’)
  36.     _end_ltp = time.time()
  37.     print(“LTP: {} <- 耗时[{}]秒”.format(lis2, (_end_ltp – _start_ltp)))

 

代码说明:

1、extract_name方法入参分别为:语句参数、类型参数。其中默认为lac模式,可以选择ltp模式。

其中lac模型提取人名的速率较快,但是ltp的提取人名准确率更高。

验证一下,执行看看效率。

Python

总结

使用的时候可以多试试两个库的区别,ltp的准确率稍微高一点。

分享:每个人都睁着眼睛,但不等于每个人都在看世界,许多人几乎不用自己的眼睛看,他们只听别人说,他们看到的世界永远是别人说的样子。——猜猜看

如果本文对你有帮助的话,点个赞吧,谢谢!

文章来自剑客·阿良

liuze
版权声明:本站原创文章,由 liuze2022-01-26发表,共计1567字。
转载说明:除特殊说明外本站文章皆由CC-4.0协议发布,转载请注明出处。
评论(没有评论)
载入中...