python机器学习---用贝叶斯算法实现垃圾邮件分类预测

yyq675886993

于 2017-07-29 17:13:26 发布

阅读量1.3k

点赞数 2

分类专栏： python机器学习文章标签： python 垃圾邮件机器学习

本文链接：https://blog.youkuaiyun.com/yyq675886993/article/details/76340366

版权

import numpy
from os import listdir
import jieba
import operator
from gensim import corpora,models,similarities
from numpy import *

#贝叶斯算法的实现
class Bayes:
    def __init__(self):
        self.length=-1
        self.labelcount=dict()
        self.vectorcount=dict()#key:value,{label:vect}
    def fit(self,dataSet,labels):
        if(len(dataSet)!=len(labels)):
            raise ValueError("您输入的类别与我们的数据集个数不匹配")
        self.length=len(dataSet[0])
        labelsnum=len(labels)#数据记录数量，类别总数量
        norelabel=set(labels)#不重复类别数组
        for item in norelabel:
            thislabel=item
            #当前类别占总类别的比例，当前类别的概率
            self.labelcount[thislabel]&#