失眠网,内容丰富有趣,生活中的好帮手!
失眠网 > 机器学习-分类算法-朴素贝叶斯算法07

机器学习-分类算法-朴素贝叶斯算法07

时间:2018-12-23 23:25:08

相关推荐

机器学习-分类算法-朴素贝叶斯算法07

概率:概率定义为一件事情发生的可能性

如:扔硬币某一面朝上有50%概率

“朴素”贝叶斯:所有特征之间条件独立

朴素贝叶斯-文档分类

from sklearn.datasets import fetch_20newsgroupsfrom sklearn.model_selection import train_test_splitfrom sklearn.feature_extraction.text import TfidfVectorizerfrom sklearn.naive_bayes import MultinomialNBdef naviebayes():# 准备数据news = fetch_20newsgroups(subset="all")print(news.data)print(news.target)# 数据分割x_train,x_test,y_train,y_test = train_test_split(news.data,news.target,test_size=0.25)# 对数据集进行特征抽取tf = TfidfVectorizer()# 以训练集当中的词的列表进行每篇文章的重要性统计x_train = tf.fit_transform(x_train)print(tf.get_feature_names())x_test = tf.transform(x_test)# 进行朴素贝叶斯算法的预测mlt = MultinomialNB(alpha=1.0)print(x_train.toarray())mlt.fit(x_train,y_train)y_predict= mlt.predict(x_test)print("预测的文章类别为:",y_predict)# 得出准确率print("准确率:",mlt.score(x_test,y_test))return Noneif __name__=="__main__":naviebayes()

如果觉得《机器学习-分类算法-朴素贝叶斯算法07》对你有帮助,请点赞、收藏,并留下你的观点哦!

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。