标签:ram text split() turn dma 求和 使用 机器 窗口
对机器学习感兴趣的小伙伴,可以借助python,实现一个N-gram分词中的Unigram和Bigram分词器,来进行入门, github地址
此项目并将前向最大切词FMM和后向最大切词的结果作为Baseline,对比分析N-gram分词器在词语切分正确率、词义消歧和新词识别等方面的优势。
本实验使用的语料是人民日报1998年中文标注的语料库,19484条。在处理过程中,按照训练集 : 测试集 = 9 : 1的比例进行随机划分。 数据预处理包括:去词性、去文本行标识(19980101-01-001-001)、词典统计、标点统计等。
首先进行语料的随机切分,默认按照9 : 1的比例切分训练集合测试集。
"""语料的随机切分,默认按照9 : 1的比例切分训练集合测试集"""
def splitCorpus(train=0.9, fileName=‘199801.txt‘):
train_file = open(‘train.txt‘, ‘wb‘)
test_file = open(‘test.txt‘, ‘wb‘)
with open(fileName, ‘rb‘) as f:
for line in f:
if random() <= train:
train_file.write(line)
else:
test_file.write(line)
train_file.close()
test_file.close()
print(‘successfully to split corpus by train = %f test = %f‘ %
(train, 1 - train))
"""统计语料词典"""
def toWordSet(file_name=‘train.txt‘, is_save=False, save_file=‘wordSet.pkl‘):
# 获取词典
word_dict = defaultdict(float)
with open(file_name, ‘rb‘) as f:
for line in f:
content = line.decode(‘gbk‘).strip().split()
# 去掉第一个词“19980101-01-001-001/m”
for word in content[1:]:
word_dict[word.split(u‘/‘)[0]] += 1
if is_save:
# 保存wordSet以复用
joblib.dump(word_dict, save_file)
print("successfully get word dictionary!")
print("the total number of words is:{0}".format(len(word_dict.keys())))
return word_dict
前向最大切词,是以可变滑动窗口对文本进行顺序取词,若改词在词典中存在,则进行一次切分;否则,缩小窗口大小,继续取词与词典库进行搜索,知道窗口词长为1。后向切词原理相似,只不过是从后面开始进行窗口滑动。
def forwardMaxCut(ustring, word_set, word_max_len=5):
"""
前向最大切词
:param ustring: 待切词文本
:param word_set: 词典
:param word_max_len: 最大词长
:return: 词列表
"""
wordList = []
if not ustring:
return wordList
while ustring:
sentence_len = len(ustring)
if sentence_len < word_max_len:
word_max_len = sentence_len
for i in range(word_max_len, 0, -1):
if ustring[:i] in word_set or i == 1:
wordList.append(ustring[:i])
ustring = ustring[i:]
break
else:
i -= 1
return wordList
运行结果:
前向分词结果:
successfully to split corpus by train = 0.900000 test = 0.100000
the total number of punction is:47
the total number of words is:53198
召回率为:0.9466013860392212
准确率为:0.9154134377927275
F值为:0.9307462195496794
后向分词结果:
successfully to split corpus by train = 0.900000 test = 0.100000
the total number of punction is:47
the total number of words is:53767
召回率为:0.950686195146746
准确率为:0.92130516483316
F值为:0.9357651113664159
由于每次运行,都会对语料进行随机切分,因此运行结果中的词典大小有出入。
Unigram切词的计算公式如下:
例句S:我是北京大学的一名研究生
# S的切分可以拆成两步
P(S) = P(我)*P(是北京大学的一名研究生)
#同时后面的子句,继续可以拆成:
P(S) = P(我)*P(是北京大学的一名研究生) = P(我)*P(是)*p(北京大学的一名研究生)
#这里有个问题,我们是如何知道应该拆成“我”和“是”两个词,而不是“我是”一个词呢
#上面计算最大概率,是递归调用的,假设我们开始有两种切分
[我,是北京大学的一名研究生]
[我是,北京大学的一名研究生]
#计算组合概率
P1 = P(我)*P(是北京大学的一名研究生)
P2 = P(我是)*P(北京大学的一名研究生)
#我们会发现
P1 > P2
#对于后面的任何子句,我们都采用无脑切分,即设置最大词长,这里假设为3,可得到以下切分:
[我,是北京大学的一名研究生]
[我是,北京大学的一名研究生]
[我是北,京大学的一名研究生]
#然后分别递归计算
#为了满足性能的要求,避免重复计算,我们采用将间接计算的子序列的组合概率,都存储起来
#每次计算新的子序列时,先查看子序列的切分组合中,是否包含已经计算过的子子序列,包含,则直接复用
def maxP(self, sentence):
‘‘‘
计算最大切分方案
:param sentence: 待切分句子
:return:
‘‘‘
# 遍历所有切分组合中,找出最大概率切分
if len(sentence) <= 1:
return self.DICT.getPValue(self, sentence)
# 判断切词方向:backward 或 forward
sentence_split_words = [self.backwardSplitSentence(
sentence), self.forwardSplitSentence(sentence)][self.split_way != ‘back‘]
# 记录最大概率值
max_p_value = 0
# 储存最大概率下的切分组合
word_pairs = []
# 组合概率值
word_p = 0
for pair in sentence_split_words:
p1, p2 = 0, 0
if pair[0] in self.value_dict:
p1 = self.value_dict[pair[0]]
else:
p1 = self.maxP(pair[0])
if pair[1] in self.value_dict:
p2 = self.value_dict[pair[1]]
else:
p2 = self.maxP(pair[1])
word_p = p1 * p2
if max_p_value < word_p:
max_p_value = word_p
word_pairs = pair
# 在词典中查询当前句对应的频率,不存在时,返回 1/N
sentence_p_value = self.DICT.getPValue(self, sentence)
# 不切分概率最大时,更新各值
if sentence_p_value > max_p_value and self.DICT.inDict(self, sentence):
self.value_dict[sentence] = sentence_p_value
self.seg_dict[sentence] = sentence
return sentence_p_value
# 某种切分组合概率最大时,更新sentence对应概率,避免后续切分重复计算
else:
self.value_dict[sentence] = max_p_value
self.seg_dict[sentence] = word_pairs
return max_p_value
运行结果:
successfully to split corpus by train = 0.900000 test = 0.100000
the total number of words is:53705
the total number of punction is:47
召回率为:0.9614382160763091
准确率为:0.9319770859102912
F值为:0.9464784466054017
例句S: 这几块地面积还真不小。
#对S进行切分,获得所有切分方案
S1 = [‘这‘, ‘几‘, ‘块‘, ‘地‘, ‘面‘, ‘积‘, ‘还‘, ‘真‘, ‘不‘, ‘小‘]
S2 = [‘这‘, ‘几‘, ‘块‘, ‘地‘, ‘面‘, ‘积‘, ‘还‘, ‘真‘, ‘不小‘]
S3 = [‘这‘, ‘几‘, ‘块‘, ‘地‘, ‘面积‘, ‘还‘, ‘真‘, ‘不‘, ‘小‘]
S4 = [‘这‘, ‘几‘, ‘块‘, ‘地‘, ‘面积‘, ‘还‘, ‘真‘, ‘不小‘]
S5 = [‘这‘, ‘几‘, ‘块‘, ‘地面‘, ‘积‘, ‘还‘, ‘真‘, ‘不小‘]
#利用Bigram公式,计算所有的方案的切分概率,为了避免出现float下溢出,采用log求和
P(S1) = -64.745
P(S2) = -63.894
P(S3) = -55.041
P(S4) = -54.190
P(S5) = -58.190
P(S4) > P(S3)>P(S5)>P(S2)>P(S1)
#不难发现,上述例句对于机器是一个歧义句,S4和S5两种切分都可以
#但是根据语境,S4是正确的
successfully to split corpus by train = 0.900000 test = 0.100000
the total number of words is:53260
The total number of bigram is : 403121.
successfully witten-Bell smoothing! smooth_value:1.3372788850370981e-05
the total number of punction is:47
召回率为:0.962036929819092
准确率为:0.9401303935308096
F值为:0.950957517059212
指标 | FMM | BMM | Unigram | Bigram |
---|---|---|---|---|
准确率 | 91.54% | 92.13% | 93.20% | 94.01% |
召回率 | 94.66% | 95.07% | 96.14% | 96.20% |
F1值 | 93.07% | 93.58% | 94.64% | 95.10% |
根据上表可知:分词效果最好的是Bigram,最差的是FMM。 因为FMM只考虑了前向顺序词是否在字典中出现,而Bigram除了考虑词典中是否包含此词,同时也考虑了邻接词对分词的选择的影响。 在处理歧义上,Bigram具有较好的效果,能基本实现消除歧义,但是消除歧义的效果受文本训练大小的影响。 在处理未登录词上,这里仅仅是对未登录词切分为单个字,因此在未登录词的处理上还要进一步的研究讨论。
标签:ram text split() turn dma 求和 使用 机器 窗口
原文地址:https://www.cnblogs.com/CocoML/p/12725988.html