利用贝叶斯算法对垃圾邮件进行分类处理

最新推荐文章于 2024-06-29 14:53:50 发布

小罗卜

最新推荐文章于 2024-06-29 14:53:50 发布

阅读量3.1k

点赞数 1

文章标签：贝叶斯分类机器学习

本文链接：https://blog.csdn.net/luoxueqian/article/details/78806984

版权

代码及注释如下：

#使用贝叶斯算法实现垃圾邮件过滤
#将一个大字符串解析为字符串列表
def textParse(bigString):
    import re
    listOfTokens = re.split(r'\W*', bigString)
    return [tok.lower() for tok in listOfTokens if len(tok) > 2]

def spamTest():
    #import pandas as pd
    docList = []; classList = []; fullText = []
    for i in range(1,26):
        #wordList = textParse(pd.read_csv('email/spam/%d.txt' %i, sep='\n', encoding='utf8'))
        wordList = textParse(open('email/spam/%d.txt' % i).read())    #spam文件夹中的邮件全设为1
        docList.append(wordList)
        fullText.extend(wordList)
        classList.append(1)
        #wordList = textParse(pd.read_csv('email/ham/%d.txt' % i, sep='\n', encoding='utf8'))
        wordList = textParse(open('email/ham/%d.txt' % i).read())    #ham文件夹中的邮件全设为0
        docList.appe