jieba分词

jieba中文分词

jieba是目前最好的Python中文分词组件,它主要有以下3种特性:

  • 支持3种分词模式
  • 支持繁体分词
  • 支持自定义词典

分词模式

分词模式:

  • 精确模式,试图将句子最精确地切开,适合文本分析;
  • 全模式,把句子中所有的可以成词的词语都扫描出来, 速度非常快,但是不能解决歧义;
  • 搜索引擎模式,在精确模式的基础上,对长词再次切分,提高召回率,适合用于搜索引擎分词。

算法操作

  • 基于前缀词典实现高效的词图扫描,生成句子中汉字所有可能成词情况所构成的有向无环图 (DAG)
  • 采用了动态规划查找最大概率路径, 找出基于词频的最大切分组合
  • 对于未登录词,采用了基于汉字成词能力的 HMM 模型,使用了 Viterbi 算法

主要功能

分词

seg_list = jieba.cut("我来到北京清华大学", cut_all=True)
print("Full Mode: " + "/ ".join(seg_list)) # 全模式
>>Full Mode:/ 来到/ 北京/ 清华/ 清华大学/ 华大/ 大学
seg_list = jieba.cut("我来到北京清华大学", cut_all=False)
print("Default Mode: " + "/ ".join(seg_list)) # 精确模式
>>Default Mode:/ 来到/ 北京/ 清华大学
seg_list = jieba.cut("他来到了网易杭研大厦")  # 默认是精确模式
print(", ".join(seg_list))
>>, 来到,, 网易, 杭研, 大厦
seg_list = jieba.cut_for_search("小明硕士毕业于中国科学院计算所,后在日本京都大学深造")  # 搜索引擎模式
print(", ".join(seg_list))
>>小明, 硕士, 毕业, 于, 中国, 科学, 学院, 科学院, 中国科学院, 计算, 计算所, ,, 后, 在, 日本, 京都, 大学, 日本京都大学, 深造
HMM模型

HMM 模型,即隐马尔可夫模型(Hidden Markov Model, HMM),是一种基于概率的统计分析模型,用来描述一个系统隐性状态的转移和隐性状态的表现概率。在 jieba 中,对于未登录到词库的词,使用了基于汉字成词能力的 HMM 模型和 Viterbi 算法,其大致原理是:

  • 采用四个隐含状态,分别表示为单字成词,词组的开头,词组的中间,词组的结尾。通过标注好的分词训练集,可以得到 HMM 的各个参数,然后使用 Viterbi 算法来解释测试集,得到分词结果。
seg_list = jieba.cut("他来到了网易杭研大厦", HMM=False)#默认精确模式和未启用HMM
print("【未启用 HMM】:" + "/ ".join(seg_list))
>>【未启用 HMM】:他/ 来到// 网易/// 大厦
seg_list = jieba.cut("他来到了网易杭研大厦", HMM=True)#默认精确模式和启用HMM
print("【识别新词】:" + "/ ".join(seg_list))
>>【识别新词】:他/ 来到// 网易/ 杭研/ 大厦

添加自定义词典

载入词典
from __future__ import print_function, unicode_literals
import sys
sys.path.append("../")
import jieba
jieba.load_userdict("E:/NLP/nlp_learn/data/userdict.txt")
import jieba.posseg as pseg

jieba.add_word('石墨烯')
jieba.add_word('凱特琳')
jieba.del_word('自定义词')

test_sent = (
"李小福是创新办主任也是云计算方面的专家; 什么是八一双鹿\n"
"例如我输入一个带“韩玉赏鉴”的标题,在自定义词库中也增加了此词为N类\n"
"「台中」正確應該不會被切開。mac上可分出「石墨烯」;此時又可以分出來凱特琳了。"
)
words = jieba.cut(test_sent)
print('/'.join(words))

print("="*40)

result = pseg.cut(test_sent)

for w in result:
    print(w.word, "/", w.flag, ", ", end=' ')

print("\n" + "="*40)

terms = jieba.cut('easy_install is great')
print('/'.join(terms))
terms = jieba.cut('python 的正则表达式是好用的')
print('/'.join(terms))

print("="*40)
# test frequency tune
testlist = [
('今天天气不错', ('今天', '天气')),
('如果放到post中将出错。', ('中', '将')),
('我们中出了一个叛徒', ('中', '出')),
]

for sent, seg in testlist:
    print('/'.join(jieba.cut(sent, HMM=False)))
    word = ''.join(seg)
    print('%s Before: %s, After: %s' % (word, jieba.get_FREQ(word), jieba.suggest_freq(seg, True)))
    print('/'.join(jieba.cut(sent, HMM=False)))
    print("-"*40)
李小福//创新办/主任///云计算/方面//专家/;/ /什么///一双/鹿/
/例如//输入/一个///韩玉赏鉴///标题///自定义/词库///增加///词为/N//
//台中//正確/應該/不會//切開//mac///分出//石墨烯///此時//可以/分出//凱特琳//========================================
李小福 / nr ,/ v ,  创新办 / i ,  主任 / b ,/ d ,/ v ,  云计算 / x ,  方面 / n ,/ uj ,  专家 / n ,  ; / x ,    / x ,  什么 / r ,/ v ,/ m ,  一双 / m ,  鹿 / nr ,  
 / x ,  例如 / v ,/ r ,  输入 / v ,  一个 / m ,/ v ,/ x ,  韩玉赏鉴 / nz ,/ x ,/ uj ,  标题 / n ,/ x ,/ p ,  自定义 / l ,  词库 / n ,/ f ,/ d ,  增加 / v ,/ ul ,/ r ,/ n ,/ p ,  N / eng ,/ q ,  
 / x ,/ x ,  台中 / s ,/ x ,  正確 / ad ,  應該 / v ,/ d ,/ v ,/ p ,  切開 / ad ,/ x ,  mac / eng ,/ f ,/ v ,  分出 / v ,/ x ,  石墨烯 / x ,/ x ,/ x ,  此時 / c ,/ d ,  可以 / c ,  分出 / v ,/ zg ,  凱特琳 / nz ,/ ul ,/ x ,  
========================================
easy_install/ /is/ /great
python/ //正则表达式//好用/========================================
今天天气/不错
今天天气 Before: 3, After: 0
今天/天气/不错
----------------------------------------
如果/放到/post/中将/出错/。
中将 Before: 763, After: 493
如果/放到/post///出错/----------------------------------------
我们////一个/叛徒
中出 Before: 3, After: 3
我们////一个/叛徒
----------------------------------------
调整词典
print('/'.join(jieba.cut('如果放到post中将出错。', HMM=False)))
>>如果/放到/post/中将/出错/
jieba.suggest_freq(('中', '将'), True)
>>490
print('/'.join(jieba.cut('如果放到post中将出错。', HMM=False)))
>>如果/放到/post///出错/
print('/'.join(jieba.cut('「台中」正确应该不会被切开', HMM=False)))
>>////正确/应该/不会//切开
jieba.suggest_freq(('台中'), True)
>>69
print('/'.join(jieba.cut('「台中」正确应该不会被切开', HMM=False)))
>>/台中//正确/应该/不会//切开
通过用户自定义词典来增强歧义纠错能力
# "大连美容美发学校中君意是你值得信赖的选择" 这句话首先会按照概率连乘最大路径来切割,因为单字有一定概率,
# 而“中君意是”这四个字中不含词典中有的词,所以会被切割成单字:
# 即:大连/ 美容美发/ 学校/ 中/ 君/ 意/ 是/ 你/ 值得/ 信赖/ 的/ 选择/
import sys
import jieba
jieba.load_userdict("E:/NLP/nlp_learn/data/userdict.txt")
print (", ".join(jieba.cut("大连美容美发学校中君意是你值得信赖的选择")))
# 解决方案是在词典中补充“君意”这个词,并给予一个词频,不用太大,比如3即可
>>大连, 美容美发, 学校,, 君意,,, 值得, 信赖,, 选择
# Bad Case: '江州/ 市/ 长江大桥/ 参加/ 了/ 长江大桥/ 的/ 通车/ 仪式'
import sys
sys.path.append("../")
import jieba
jieba.load_userdict("E:/NLP/nlp_learn/data/userdict.txt")
print (", ".join(jieba.cut("江州市长江大桥参加了长江大桥的通车仪式")))
# 我是这样设置的:

# ==user.dict==

# 江大桥 20000
江州, 市长, 江大桥, 参加,, 长江大桥,, 通车, 仪式

关键词提取

基于TF-IDF算法的关键词抽取

TF-IDF(Term Frequency-Inverse Document Frequency, 词频-逆文件频率)是一种统计方法,用以评估一个词语对于一个文件集或一个语料库中的一份文件的重要程度,其原理可概括为:

  • 一个词语在一篇文章中出现次数越多,同时在所有文档中出现次数越少,越能够代表该文章

方法:jieba.analyse.extract_tags,该方法有4个参数:

  • sentence:为待提取的文本
  • topK:为返回几个 TF/IDF 权重最大的关键词,默认值为 20
  • withWeight:是否一并返回关键词权重值,默认值为 False
  • allowPOS:仅包括指定词性的词,默认值为空
import jieba.analyse
s = "此外,公司拟对全资子公司吉林欧亚置业有限公司增资4.3亿元,增资后,吉林欧亚置业注册资本由7000万元增加到5亿元。吉林欧亚置业主要经营范围为房地产开发及百货零售等业务。目前在建吉林欧亚城市商业综合体项目。2013年,实现营业收入0万元,实现净利润-139.13万元。"
for x, w in jieba.analyse.extract_tags(s, topK=20, withWeight=True):
    print('%s %s' % (x, w))
>>欧亚 0.7300142700289363
吉林 0.659038184373617
置业 0.4887134522112766
万元 0.3392722481859574
增资 0.33582401985234045
4.3 0.25435675538085106
7000 0.25435675538085106
2013 0.25435675538085106
139.13 0.25435675538085106
实现 0.19900979900382978
综合体 0.19480309624702127
经营范围 0.19389757253595744
亿元 0.1914421623587234
在建 0.17541884768425534
全资 0.17180164988510638
注册资本 0.1712441526
百货 0.16734460041382979
零售 0.1475057117057447
子公司 0.14596045237787234
营业 0.13920178509021275
基于 TextRank 算法的关键词抽取

基本思想:

  1. 将待抽取关键词的文本进行分词
  2. 以固定窗口大小(默认为5,通过span属性调整),词之间的共现关系,构建图
  3. 计算图中节点的PageRank,注意是无向带权图

TextRank 是另一种基于PageRank关键词提取算法。通过 jieba.analyse.textrank 方法可以使用基于 TextRank 算法的关键词提取,其与 ‘jieba.analyse.extract_tags’ 有一样的参数,但前者默认过滤词性(allowPOS=(‘ns’, ‘n’, ‘vn’, ‘v’))。

for x, w in jieba.analyse.textrank(s, withWeight=True):
    print('%s %s' % (x, w))
>>吉林 1.0
欧亚 0.9966893354178172
置业 0.6434360313092776
实现 0.5898606692859626
收入 0.43677859947991454
增资 0.4099900531283276
子公司 0.35678295947672795
城市 0.34971383667403655
商业 0.34817220716026936
业务 0.3092230992619838
在建 0.3077929164033088
营业 0.3035777049319588
全资 0.303540981053475
综合体 0.29580869172394825
注册资本 0.29000519464085045
有限公司 0.2807830798576574
零售 0.27883620861218145
百货 0.2781657628445476
开发 0.2693488779295851
经营范围 0.2642762173558316
自定义语料库

关键词提取所使用逆向文件频率(IDF)文本语料库和停止词(Stop Words)文本语料库可以切换成自定义语料库的路径。

jieba.analyse.set_stop_words("stop_words.txt")
jieba.analyse.set_idf_path("idf.txt.big");
for x, w in anls.extract_tags(s, topK=20, withWeight=True):
    print('%s %s' % (x, w))
>>吉林 1.0174270215234043
欧亚 0.7300142700289363
增资 0.5087135107617021
实现 0.5087135107617021
置业 0.4887134522112766
万元 0.3392722481859574
此外 0.25435675538085106
全资 0.25435675538085106
有限公司 0.25435675538085106
4.3 0.25435675538085106
注册资本 0.25435675538085106
7000 0.25435675538085106
增加 0.25435675538085106
主要 0.25435675538085106
房地产 0.25435675538085106
业务 0.25435675538085106
目前 0.25435675538085106
城市 0.25435675538085106
综合体 0.25435675538085106
2013 0.25435675538085106

词性标注

jieba.posseg.POSTokenizer(tokenizer=None) 新建自定义分词器,tokenizer 参数可指定内部使用的 jieba.Tokenizer 分词器。jieba.posseg.dt 为默认词性标注分词器。

import jieba.posseg as pseg
words = pseg.cut("我爱北京天安门")
for word, flag in words:
    print('%s %s' % (word, flag))
>>
我 r
爱 v
北京 ns
天安门 ns

并行分词

将目标文本按行分隔后,把各行文本分配到多个 Python 进程并行分词,然后归并结果,从而获得分词速度的可观提升。用法:

  • jieba.enable_parallel(4):开启并行分词模式,参数为并行进程数
  • jieba.disable_parallel() :关闭并行分词模式
注意:基于 python 自带的 multiprocessing 模块,目前暂不支持 Windows

返回词语在原文的起止位置

# 默认模式
result = jieba.tokenize(u"永和服装饰品有限公司")
for tk in result:
    print("word %s\t\t start: %d \t\t end:%d" % (tk[0],tk[1],tk[2]))
>>
word 永和		 start: 0 		 end:2
word 服装		 start: 2 		 end:4
word 饰品		 start: 4 		 end:6
word 有限公司		 start: 6 		 end:10
# 搜索模式
result = jieba.tokenize(u'永和服装饰品有限公司', mode='search')
for tk in result:
    print("word %s\t\t start: %d \t\t end:%d" % (tk[0],tk[1],tk[2]))
>>
word 永和		 start: 0 		 end:2
word 服装		 start: 2 		 end:4
word 饰品		 start: 4 		 end:6
word 有限		 start: 6 		 end:8
word 公司		 start: 8 		 end:10
word 有限公司		 start: 6 		 end:10

参考资料

结巴中文分词
简明jieba中文分词教程

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值