朴素贝叶斯

最新推荐文章于 2024-12-13 00:04:09 发布

跟我回xwy

最新推荐文章于 2024-12-13 00:04:09 发布

阅读量134

点赞数

文章标签：朴素贝叶斯 python 分类

本文链接：https://blog.youkuaiyun.com/qq_38728337/article/details/83070217

版权

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Mon Oct 15 18:15:01 2018

@author: xiaweiyi
"""
from sklearn.datasets import fetch_20newsgroups
news=fetch_20newsgroups(subset='all')
#print(len(news.data))
#print(news.data[0])

from sklearn.model_selection import train_test_split
X_train,X_test,y_train,y_test=train_test_split(news.data,news.target,test_size=0.25,random_state=33)

from sklearn.feature_extraction.text import CountVectorizer
vec = CountVectorizer()
X_train=vec.fit_transform(X_train)
X_test=vec.transform(X_test)

from sklearn.naive_bayes import MultinomialNB
mnb=MultinomialNB()
mnb.fit(X_train,y_train)
y_predict=mnb.predict(X_test)

from sklearn.metrics import classification_report
print('The accuracy of Naive Bayes Classifier is',mnb.score(X_test,y_test))
print(classification_report(y_test,y_predict,target_names=news.target_names))