ML之NB:基于NB朴素贝叶斯算法训练20类新闻文本数据集进行多分类预测

欢迎光临程序代写小店https://item.taobao.com/item.htm?spm=a230r.1.14.59.255028c3ALNkZ0&id=586797758241&ns=1&abbucket=15#detail

matlab程序代写C++代做C语言java安卓app设计python网站php代码 一流985硕博团队为您服务

 

ML之NB:基于NB朴素贝叶斯算法训练20类新闻文本数据集进行多分类预测

输出结果

ML之NB:基于NB朴素贝叶斯算法训练20类新闻文本数据集进行多分类预测

设计思路

ML之NB:基于NB朴素贝叶斯算法训练20类新闻文本数据集进行多分类预测
核心代码

vec = CountVectorizer()
X_train = vec.fit_transform(X_train)
X_test = vec.transform(X_test)
 
 
mnb = MultinomialNB()
mnb.fit(X_train, y_train)         
y_predict = mnb.predict(X_test)  
 
 
print('The accuracy of Naive Bayes Classifier is', mnb.score(X_test, y_test))