北大開源中文分詞工具包 pkuseg

本文作者：汪思穎

2019-01-10 16:21

導語：這一工具具有如下三個特點：高分詞準確率、多領域分詞、支持用戶自訓練模型。

雷鋒網(wǎng) AI 科技評論消息，日前，北京大學語言計算與機器學習研究組研制推出一套全新中文分詞工具包 pkuseg，這一工具包有如下三個特點：

高分詞準確率。相比于其他的分詞工具包，當使用相同的訓練數(shù)據(jù)和測試數(shù)據(jù)，pkuseg 可以取得更高的分詞準確率。
多領域分詞。不同于以往的通用中文分詞工具，此工具包同時致力于為不同領域的數(shù)據(jù)提供個性化的預訓練模型。根據(jù)待分詞文本的領域特點，用戶可以自由地選擇不同的模型。而其他現(xiàn)有分詞工具包，一般僅提供通用領域模型。
支持用戶自訓練模型。支持用戶使用全新的標注數(shù)據(jù)進行訓練。

各項性能對比如下：

與 jieba、THULAC 等國內代表分詞工具包進行性能比較：

考慮到 jieba 分詞和 THULAC 工具包等并沒有提供細領域的預訓練模型，為了便于比較，開發(fā)團隊重新使用它們提供的訓練接口在細領域的數(shù)據(jù)集上進行訓練，用訓練得到的模型進行中文分詞。他們選擇 Linux 作為測試環(huán)境，在新聞數(shù)據(jù)(MSRA)、混合型文本(CTB8)、網(wǎng)絡文本(WEIBO)數(shù)據(jù)上對不同工具包進行了準確率測試。在此過程中，他們使用第二屆國際漢語分詞評測比賽提供的分詞評價腳本，其中 MSRA 與 WEIBO 使用標準訓練集測試集劃分，CTB8 采用隨機劃分。對于不同的分詞工具包，訓練測試數(shù)據(jù)的劃分都是一致的；即所有的分詞工具包都在相同的訓練集上訓練，在相同的測試集上測試。

以下是在不同數(shù)據(jù)集上的對比結果：

北大開源中文分詞工具包 pkuseg

同時，為了比較細領域分詞的優(yōu)勢，開發(fā)團隊比較了他們的方法和通用分詞模型的效果對比。其中 jieba 和 THULAC 均使用了軟件包提供的、默認的分詞模型：

北大開源中文分詞工具包 pkuseg 從結果上來看，當用戶了解待分詞文本的領域時，細領域分詞可以取得更好的效果。然而 jieba 和 THULAC 等分詞工具包僅提供了通用領域模型。

目前，該工具包已經(jīng)在 GitHub 開源，編譯、安裝和使用說明如下。

編譯和安裝

1. 通過 pip 下載(自帶模型文件)

pip install pkuseg
之后通過 import pkuseg 來引用

2. 從 github 下載(需要下載模型文件，見預訓練模型)

將 pkuseg 文件放到目錄下，通過 import pkuseg 使用
模型需要下載或自己訓練。

使用方式

1. 代碼示例

代碼示例1 使用默認模型及默認詞典分詞
import pkuseg
seg = pkuseg.pkuseg() #以默認配置加載模型
text = seg.cut('我愛北京天安門') #進行分詞
print(text)

代碼示例2 設置用戶自定義詞典
import pkuseg
lexicon = ['北京大學', '北京天安門'] #希望分詞時用戶詞典中的詞固定不分開
seg = pkuseg.pkuseg(user_dict=lexicon) #加載模型，給定用戶詞典
text = seg.cut('我愛北京天安門') #進行分詞
print(text)

代碼示例3
import pkuseg
seg = pkuseg.pkuseg(model_name='./ctb8') #假設用戶已經(jīng)下載好了ctb8的模型并放在了'./ctb8'目錄下，通過設置model_name加載該模型
text = seg.cut('我愛北京天安門') #進行分詞
print(text)

代碼示例4
import pkuseg
pkuseg.test('input.txt', 'output.txt', nthread=20) #對input.txt的文件分詞輸出到output.txt中，使用默認模型和詞典，開20個進程

代碼示例5
import pkuseg
pkuseg.train('msr_training.utf8', 'msr_test_gold.utf8', './models', nthread=20) #訓練文件為'msr_training.utf8'，測試文件為'msr_test_gold.utf8'，模型存到'./models'目錄下，開20個進程訓練模型

2. 參數(shù)說明

pkuseg.pkuseg(model_name='ctb8', user_dict=[])
model_name 模型路徑。默認是'ctb8'表示我們預訓練好的模型(僅對pip下載的用戶)。用戶可以填自己下載或訓練的模型所在的路徑如model_name='./models'。
user_dict 設置用戶詞典。默認不使用詞典。填'safe_lexicon'表示我們提供的一個中文詞典(僅pip)。用戶可以傳入一個包含若干自定義單詞的迭代器。

pkuseg.test(readFile, outputFile, model_name='ctb8', user_dict=[], nthread=10)
readFile 輸入文件路徑
outputFile 輸出文件路徑
model_name 同pkuseg.pkuseg
user_dict 同pkuseg.pkuseg
nthread 測試時開的進程數(shù)