此处将结果输出到result.txt文件中
各词概率保存到wordsProb.txt中
Code:def getProbWord(self, testDict, normalDict, spamDict, numNormal, numSpam):
"""
计算对分类结果影响最大的15个词
:param testDict: 测试数据字典
:param normalDict: 正常邮件字典
:param spamDict: 垃圾邮件字典
:param numNormal: 正常邮件的数量
:param numSpam: 垃圾邮件的数量
:return wordProbList: 对分类结果影响最大的15个词
"""
wordProbList = {}
for word, num in testDict.items():
# 当词不在垃圾邮件词表中,在正常邮件词表中,计算概率
if word in spamDict.keys() and word in normalDict.keys():
# 求类先验概率
# 正常邮件
pw_n = normalDict[word] / numNormal
# 垃圾邮件
pw_s = spamDict[word] / numSpam
ps_w = pw_s / (pw_s + pw_n)
wordProbList[word] = ps_w
# 当词在垃圾邮件词表中,不在正常邮件词表中,计算概率
if wo