首页
学习
活动
专区
工具
TVP
发布
社区首页 >问答首页 >通过CountVectorizer 和chi2特征提取进行文本分类 准确率只有0.348正常吗?

通过CountVectorizer 和chi2特征提取进行文本分类 准确率只有0.348正常吗?

提问于 2019-04-23 13:22:03
回答 0关注 0查看 617
代码语言:javascript
复制
#encoding=utf-8

from sklearn.feature_selection import SelectKBest
from sklearn.feature_selection import chi2
from sklearn.datasets import fetch_20newsgroups
from sklearn.feature_extraction.text import CountVectorizer
from sklearn.naive_bayes import MultinomialNB
from sklearn.svm import  SVC
from sklearn.model_selection import GridSearchCV
from sklearn.metrics import accuracy_score,make_scorer
from sklearn.metrics import classification_report,accuracy_score
import numpy as n

categories = ['alt.atheism','talk.religion.misc','comp.graphics','sci.space']
data_train = fetch_20newsgroups(subset='train',categories=categories)
data_test = fetch_20newsgroups(subset='test',categories=categories)

count = CountVectorizer(stop_words='english',min_df=2)
train_feature = count.fit_transform(data_train.data)
print train_feature.toarray()
chi = SelectKBest(chi2,k=500)
train = chi.fit_transform(train_feature,data_train.target)
test_feature = count.fit_transform(data_test.data)
test = chi.fit_transform(test_feature,data_test.target)
# clf = MultinomialNB()
# classifier = clf.fit(train,data_train.target)
# test_accuracy = classifier.score(test,data_test.target)
# print test_accuracy

svc = SVC()
param_grid = {'kernel':('linear','poly'),'C':[0.1,1]}
grid_search = GridSearchCV(svc,param_grid=param_grid,cv=5,scoring=make_scorer(accuracy_score))
grid_search.fit(train,data_train.target)
y = grid_search.predict(test)
print grid_search.best_estimator_
print 'accuracy',accuracy_score(data_test.target,y)
print classification_report(data_test.target,y)

结果:

accuracy 0.34885439763488546

precision recall f1-score support

0 0.34 0.39 0.36 319

1 0.42 0.43 0.43 389

2 0.36 0.33 0.34 394

3 0.22 0.20 0.21 251

micro avg 0.35 0.35 0.35 1353

macro avg 0.34 0.34 0.34 1353

weighted avg 0.35 0.35 0.35 1353

希望有大佬可以指点一下,哪有问题呢

回答

和开发者交流更多问题细节吧,去 写回答
相关文章

相似问题

相关问答用户
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档