Compare commits

...

7 Commits

Author SHA1 Message Date
Neutrino
67fa2e36e7
Update README.md update paddle link. (#817) 2020-02-15 16:33:35 +08:00
fxsjy
1e20c89b66 fix setup.py in python2.7 2020-01-20 22:22:34 +08:00
fxsjy
5704e23bbf update version: 0.42 2020-01-13 21:24:45 +08:00
fxsjy
aa65031788 fix file mode 2020-01-13 21:03:38 +08:00
fxsjy
2eb11c8028 fix issue #810 2020-01-13 20:53:43 +08:00
JesseyXujin
d703bce302 paddle coredump exception fix (#807)
* paddle_null_point_fix

* add core expception note

* delete yield

* modify test paddle for supporting enable_paddle()
2020-01-10 16:30:46 +08:00
vissssa
dc2b788eb3 refactor: improvement check_paddle_installed (#806) 2020-01-09 19:23:11 +08:00
37 changed files with 73 additions and 74 deletions

View File

@ -1,3 +1,11 @@
2019-1-20: version 0.42.1
1. 修复setup.py在python2.7版本无法工作的问题 (issue #809)
2019-1-13: version 0.42
1. 修复paddle模式空字符串coredump问题 @JesseyXujin
2. 修复cut_all模式切分丢字问题 @fxsjy
3. paddle安装检测优化 @vissssa
2019-1-8: version 0.41 2019-1-8: version 0.41
1. 开启paddle模式更友好 1. 开启paddle模式更友好
2. 修复cut_all模式不支持中英混合词的bug 2. 修复cut_all模式不支持中英混合词的bug

View File

@ -13,7 +13,7 @@ jieba
* 精确模式,试图将句子最精确地切开,适合文本分析; * 精确模式,试图将句子最精确地切开,适合文本分析;
* 全模式,把句子中所有的可以成词的词语都扫描出来, 速度非常快,但是不能解决歧义; * 全模式,把句子中所有的可以成词的词语都扫描出来, 速度非常快,但是不能解决歧义;
* 搜索引擎模式,在精确模式的基础上,对长词再次切分,提高召回率,适合用于搜索引擎分词。 * 搜索引擎模式,在精确模式的基础上,对长词再次切分,提高召回率,适合用于搜索引擎分词。
* paddle模式利用PaddlePaddle深度学习框架训练序列标注双向GRU网络模型实现分词。同时支持词性标注。paddle模式使用需安装paddlepaddle-tiny`pip install paddlepaddle-tiny==1.6.1`。目前paddle模式支持jieba v0.40及以上版本。jieba v0.40以下版本请升级jieba`pip install jieba --upgrade` 。[PaddlePaddle官网](www.paddlepaddle.org.cn) * paddle模式利用PaddlePaddle深度学习框架训练序列标注双向GRU网络模型实现分词。同时支持词性标注。paddle模式使用需安装paddlepaddle-tiny`pip install paddlepaddle-tiny==1.6.1`。目前paddle模式支持jieba v0.40及以上版本。jieba v0.40以下版本请升级jieba`pip install jieba --upgrade` 。[PaddlePaddle官网](https://www.paddlepaddle.org.cn/)
* 支持繁体分词 * 支持繁体分词
* 支持自定义词典 * 支持自定义词典
* MIT 授权协议 * MIT 授权协议

View File

@ -1,26 +1,24 @@
from __future__ import absolute_import, unicode_literals from __future__ import absolute_import, unicode_literals
__version__ = '0.41'
__version__ = '0.42.1'
__license__ = 'MIT' __license__ = 'MIT'
import re
import os
import sys
import time
import logging
import marshal import marshal
import re
import tempfile import tempfile
import threading import threading
from math import log import time
from hashlib import md5 from hashlib import md5
from ._compat import * from math import log
from . import finalseg from . import finalseg
from ._compat import *
if os.name == 'nt': if os.name == 'nt':
from shutil import move as _replace_file from shutil import move as _replace_file
else: else:
_replace_file = os.rename _replace_file = os.rename
_get_abs_path = lambda path: os.path.normpath(os.path.join(os.getcwd(), path)) _get_abs_path = lambda path: os.path.normpath(os.path.join(os.getcwd(), path))
DEFAULT_DICT = None DEFAULT_DICT = None
@ -47,10 +45,11 @@ re_han_default = re.compile("([\u4E00-\u9FD5a-zA-Z0-9+#&\._%\-]+)", re.U)
re_skip_default = re.compile("(\r\n|\s)", re.U) re_skip_default = re.compile("(\r\n|\s)", re.U)
def setLogLevel(log_level): def setLogLevel(log_level):
global logger
default_logger.setLevel(log_level) default_logger.setLevel(log_level)
class Tokenizer(object): class Tokenizer(object):
def __init__(self, dictionary=DEFAULT_DICT): def __init__(self, dictionary=DEFAULT_DICT):
@ -69,7 +68,8 @@ class Tokenizer(object):
def __repr__(self): def __repr__(self):
return '<Tokenizer dictionary=%r>' % self.dictionary return '<Tokenizer dictionary=%r>' % self.dictionary
def gen_pfdict(self, f): @staticmethod
def gen_pfdict(f):
lfreq = {} lfreq = {}
ltotal = 0 ltotal = 0
f_name = resolve_filename(f) f_name = resolve_filename(f)
@ -205,14 +205,15 @@ class Tokenizer(object):
eng_scan = 0 eng_scan = 0
yield eng_buf yield eng_buf
if len(L) == 1 and k > old_j: if len(L) == 1 and k > old_j:
if re_eng.match(sentence[k]): word = sentence[k:L[0] + 1]
if re_eng.match(word):
if eng_scan == 0: if eng_scan == 0:
eng_scan = 1 eng_scan = 1
eng_buf = sentence[k] eng_buf = word
else: else:
eng_buf += sentence[k] eng_buf += word
if eng_scan == 0: if eng_scan == 0:
yield sentence[k:L[0] + 1] yield word
old_j = L[0] old_j = L[0]
else: else:
for j in L: for j in L:
@ -286,7 +287,7 @@ class Tokenizer(object):
yield elem yield elem
def cut(self, sentence, cut_all=False, HMM=True, use_paddle=False): def cut(self, sentence, cut_all=False, HMM=True, use_paddle=False):
''' """
The main function that segments an entire sentence that contains The main function that segments an entire sentence that contains
Chinese characters into separated words. Chinese characters into separated words.
@ -294,14 +295,12 @@ class Tokenizer(object):
- sentence: The str(unicode) to be segmented. - sentence: The str(unicode) to be segmented.
- cut_all: Model type. True for full pattern, False for accurate pattern. - cut_all: Model type. True for full pattern, False for accurate pattern.
- HMM: Whether to use the Hidden Markov Model. - HMM: Whether to use the Hidden Markov Model.
''' """
is_paddle_installed = False is_paddle_installed = check_paddle_install['is_paddle_installed']
if use_paddle == True:
is_paddle_installed = check_paddle_install()
sentence = strdecode(sentence) sentence = strdecode(sentence)
if use_paddle == True and is_paddle_installed == True: if use_paddle and is_paddle_installed:
if sentence is None or sentence == "" or sentence == u"": # if sentence is null, it will raise core exception in paddle.
yield sentence if sentence is None or len(sentence) == 0:
return return
import jieba.lac_small.predict as predict import jieba.lac_small.predict as predict
results = predict.get_sent(sentence) results = predict.get_sent(sentence)

View File

@ -1,18 +1,22 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
import logging
import os import os
import sys import sys
import logging
log_console = logging.StreamHandler(sys.stderr) log_console = logging.StreamHandler(sys.stderr)
default_logger = logging.getLogger(__name__) default_logger = logging.getLogger(__name__)
default_logger.setLevel(logging.DEBUG) default_logger.setLevel(logging.DEBUG)
def setLogLevel(log_level): def setLogLevel(log_level):
global logger
default_logger.setLevel(log_level) default_logger.setLevel(log_level)
check_paddle_install = {'is_paddle_installed': False}
try: try:
import pkg_resources import pkg_resources
get_module_res = lambda *res: pkg_resources.resource_stream(__name__, get_module_res = lambda *res: pkg_resources.resource_stream(__name__,
os.path.join(*res)) os.path.join(*res))
except ImportError: except ImportError:
@ -21,7 +25,6 @@ except ImportError:
def enable_paddle(): def enable_paddle():
import_paddle_check = False
try: try:
import paddle import paddle
except ImportError: except ImportError:
@ -30,21 +33,24 @@ def enable_paddle():
try: try:
import paddle import paddle
except ImportError: except ImportError:
default_logger.debug("Import paddle error, please use command to install: pip install paddlepaddle-tiny==1.6.1." default_logger.debug(
"Import paddle error, please use command to install: pip install paddlepaddle-tiny==1.6.1."
"Now, back to jieba basic cut......") "Now, back to jieba basic cut......")
if paddle.__version__ < '1.6.1': if paddle.__version__ < '1.6.1':
default_logger.debug("Find your own paddle version doesn't satisfy the minimum requirement (1.6.1), " default_logger.debug("Find your own paddle version doesn't satisfy the minimum requirement (1.6.1), "
"please install paddle tiny by 'pip install --upgrade paddlepaddle-tiny', " "please install paddle tiny by 'pip install --upgrade paddlepaddle-tiny', "
"or upgrade paddle full version by 'pip install --upgrade paddlepaddle (-gpu for GPU version)' ") "or upgrade paddle full version by "
"'pip install --upgrade paddlepaddle (-gpu for GPU version)' ")
else: else:
try: try:
import jieba.lac_small.predict as predict import jieba.lac_small.predict as predict
import_paddle_check = True
default_logger.debug("Paddle enabled successfully......") default_logger.debug("Paddle enabled successfully......")
check_paddle_install['is_paddle_installed'] = True
except ImportError: except ImportError:
default_logger.debug("Import error, cannot find paddle.fluid and jieba.lac_small.predict module. " default_logger.debug("Import error, cannot find paddle.fluid and jieba.lac_small.predict module. "
"Now, back to jieba basic cut......") "Now, back to jieba basic cut......")
PY2 = sys.version_info[0] == 2 PY2 = sys.version_info[0] == 2
default_encoding = sys.getfilesystemencoding() default_encoding = sys.getfilesystemencoding()
@ -66,6 +72,7 @@ else:
itervalues = lambda d: iter(d.values()) itervalues = lambda d: iter(d.values())
iteritems = lambda d: iter(d.items()) iteritems = lambda d: iter(d.items())
def strdecode(sentence): def strdecode(sentence):
if not isinstance(sentence, text_type): if not isinstance(sentence, text_type):
try: try:
@ -74,25 +81,9 @@ def strdecode(sentence):
sentence = sentence.decode('gbk', 'ignore') sentence = sentence.decode('gbk', 'ignore')
return sentence return sentence
def resolve_filename(f): def resolve_filename(f):
try: try:
return f.name return f.name
except AttributeError: except AttributeError:
return repr(f) return repr(f)
def check_paddle_install():
is_paddle_installed = False
try:
import paddle
if paddle.__version__ >= '1.6.1':
is_paddle_installed = True
else:
is_paddle_installed = False
default_logger.debug("Check the paddle version is not correct, the current version is "+ paddle.__version__+","
"please use command to install paddle: pip uninstall paddlepaddle(-gpu), "
"pip install paddlepaddle-tiny==1.6.1. Now, back to jieba basic cut......")
except ImportError:
default_logger.debug("Import paddle error, back to jieba basic cut......")
is_paddle_installed = False
return is_paddle_installed

0
jieba/lac_small/__init__.py Executable file → Normal file
View File

0
jieba/lac_small/creator.py Executable file → Normal file
View File

0
jieba/lac_small/model_baseline/crfw Executable file → Normal file
View File

0
jieba/lac_small/model_baseline/fc_0.b_0 Executable file → Normal file
View File

0
jieba/lac_small/model_baseline/fc_0.w_0 Executable file → Normal file
View File

0
jieba/lac_small/model_baseline/fc_1.b_0 Executable file → Normal file
View File

0
jieba/lac_small/model_baseline/fc_1.w_0 Executable file → Normal file
View File

0
jieba/lac_small/model_baseline/fc_2.b_0 Executable file → Normal file
View File

0
jieba/lac_small/model_baseline/fc_2.w_0 Executable file → Normal file
View File

0
jieba/lac_small/model_baseline/fc_3.b_0 Executable file → Normal file
View File

0
jieba/lac_small/model_baseline/fc_3.w_0 Executable file → Normal file
View File

0
jieba/lac_small/model_baseline/fc_4.b_0 Executable file → Normal file
View File

0
jieba/lac_small/model_baseline/fc_4.w_0 Executable file → Normal file
View File

0
jieba/lac_small/model_baseline/gru_0.b_0 Executable file → Normal file
View File

0
jieba/lac_small/model_baseline/gru_0.w_0 Executable file → Normal file
View File

0
jieba/lac_small/model_baseline/gru_1.b_0 Executable file → Normal file
View File

0
jieba/lac_small/model_baseline/gru_1.w_0 Executable file → Normal file
View File

0
jieba/lac_small/model_baseline/gru_2.b_0 Executable file → Normal file
View File

0
jieba/lac_small/model_baseline/gru_2.w_0 Executable file → Normal file
View File

0
jieba/lac_small/model_baseline/gru_3.b_0 Executable file → Normal file
View File

0
jieba/lac_small/model_baseline/gru_3.w_0 Executable file → Normal file
View File

0
jieba/lac_small/model_baseline/word_emb Executable file → Normal file
View File

0
jieba/lac_small/nets.py Executable file → Normal file
View File

0
jieba/lac_small/predict.py Executable file → Normal file
View File

0
jieba/lac_small/reader_small.py Executable file → Normal file
View File

0
jieba/lac_small/tag.dic Executable file → Normal file
View File

0
jieba/lac_small/utils.py Executable file → Normal file
View File

0
jieba/lac_small/word.dic Executable file → Normal file
View File

View File

@ -1,11 +1,11 @@
from __future__ import absolute_import, unicode_literals from __future__ import absolute_import, unicode_literals
import os
import re
import sys
import jieba
import pickle import pickle
from .._compat import * import re
import jieba
from .viterbi import viterbi from .viterbi import viterbi
from .._compat import *
PROB_START_P = "prob_start.p" PROB_START_P = "prob_start.p"
PROB_TRANS_P = "prob_trans.p" PROB_TRANS_P = "prob_trans.p"
@ -252,6 +252,7 @@ class POSTokenizer(object):
def lcut(self, *args, **kwargs): def lcut(self, *args, **kwargs):
return list(self.cut(*args, **kwargs)) return list(self.cut(*args, **kwargs))
# default Tokenizer instance # default Tokenizer instance
dt = POSTokenizer(jieba.dt) dt = POSTokenizer(jieba.dt)
@ -276,12 +277,10 @@ def cut(sentence, HMM=True, use_paddle=False):
Note that this only works using dt, custom POSTokenizer Note that this only works using dt, custom POSTokenizer
instances are not supported. instances are not supported.
""" """
is_paddle_installed = False is_paddle_installed = check_paddle_install['is_paddle_installed']
if use_paddle == True: if use_paddle and is_paddle_installed:
is_paddle_installed = check_paddle_install() # if sentence is null, it will raise core exception in paddle.
if use_paddle==True and is_paddle_installed == True:
if sentence is None or sentence == "" or sentence == u"": if sentence is None or sentence == "" or sentence == u"":
yield pair(None, None)
return return
import jieba.lac_small.predict as predict import jieba.lac_small.predict as predict
sents, tags = predict.get_result(strdecode(sentence)) sents, tags = predict.get_result(strdecode(sentence))

View File

@ -43,7 +43,7 @@ GitHub: https://github.com/fxsjy/jieba
""" """
setup(name='jieba', setup(name='jieba',
version='0.41', version='0.42.1',
description='Chinese Words Segmentation Utilities', description='Chinese Words Segmentation Utilities',
long_description=LONGDOC, long_description=LONGDOC,
author='Sun, Junyi', author='Sun, Junyi',
@ -71,5 +71,5 @@ setup(name='jieba',
keywords='NLP,tokenizing,Chinese word segementation', keywords='NLP,tokenizing,Chinese word segementation',
packages=['jieba'], packages=['jieba'],
package_dir={'jieba':'jieba'}, package_dir={'jieba':'jieba'},
package_data={'jieba':['*.*','finalseg/*','analyse/*','posseg/*', 'lac_small/*','lac_small/model_baseline/*']} package_data={'jieba':['*.*','finalseg/*','analyse/*','posseg/*', 'lac_small/*.py','lac_small/*.dic', 'lac_small/model_baseline/*']}
) )

View File

@ -98,3 +98,4 @@ if __name__ == "__main__":
cuttest('你认识那个和主席握手的的哥吗?他开一辆黑色的士。') cuttest('你认识那个和主席握手的的哥吗?他开一辆黑色的士。')
jieba.add_word('超敏C反应蛋白') jieba.add_word('超敏C反应蛋白')
cuttest('超敏C反应蛋白是什么, java好学吗?,小潘老板都学Python') cuttest('超敏C反应蛋白是什么, java好学吗?,小潘老板都学Python')
cuttest('steel健身爆发力运动兴奋补充剂')

View File

@ -2,7 +2,7 @@
import sys import sys
sys.path.append("../") sys.path.append("../")
import jieba import jieba
jieba.enable_paddle()
def cuttest(test_sent): def cuttest(test_sent):
result = jieba.cut(test_sent, use_paddle=True) result = jieba.cut(test_sent, use_paddle=True)

View File

@ -2,7 +2,8 @@
import sys import sys
sys.path.append("../") sys.path.append("../")
import jieba.posseg as pseg import jieba.posseg as pseg
import jieba
jieba.enable_paddle()
def cuttest(test_sent): def cuttest(test_sent):
result = pseg.cut(test_sent, use_paddle=True) result = pseg.cut(test_sent, use_paddle=True)