贝叶斯算法实例源代码.pdfVIP

臣心一片磁针石，不指南方不肯休。——文天祥

贝叶斯算法实例源代码

以下是一个基于贝叶斯算法的垃圾邮件分类的实例源代码：

python

importos

importstring

fromcollectionsimportCounter

classSpamFilter:

def__init__(self,spam_dir,ham_dir):

self.spam_dir=spam_dir

self.ham_dir=ham_dir

臣心一片磁针石，不指南方不肯休。——文天祥

self.spam_words=Counter()

self.ham_words=Counter()

self.spam_total_words=0

self.ham_total_words=0

self.p_spam=0

self.p_ham=0

self.train()

deftrain(self):

#训练

forpathinos.listdir(self.spam_dir):

臣心一片磁针石，不指南方不肯休。——文天祥

withopen(os.path.join(self.spam_dir,path),r,

encoding=ISO-8859-1)asf:

forlineinf.readlines():

string.punctuation)).split()

words=[w.lower()forwinwords]

self.spam_words.update(words)

self.spam_total_words+=len(words)

forpathinos.listdir(self.ham_dir):

withopen(os.path.join(self.ham_dir,path),r,

encoding=ISO-8859-1)asf:

forlineinf.readlines():

臣心一片磁针石，不指南方不肯休。——文天祥

string.punctuation)).split()

words=[w.lower()forwinwords]

self.ham_words.update(words)

self.ham_total_words+=len(words)

self.p_spam=l