SkLearn Multinomial NB:提取所有功能

时间:2017-03-15 15:52:20

标签: python feature-extraction

我使用MultinomialNB()分类器和TfidfVectorizer来构建二进制模型(只有2个类) 我想打印并保存到csv文件中的coef_,功能和标签的完整列表。 会发生什么是我只能打印一个类的功能 - 一旦我尝试打印第二个类就会出错。我理解这是因为类是二进制的? 我使用了以下代码

指数= 0 coef_features_c1_c2 = []

for feat,c1,c2 in zip(vector.get_feature_names(),clf.feature_count_ [0],clf.feature_count_ [1]):     coef_features_c1_c2.append(tuple([clf.coef_ [0] [index],feat,c1,c2]))     索引+ = 1

for i in sorted(coef_features_c1_c2):     print(i)

How to get most informative features for scikit-learn classifier for different class?

我的问题是如何获得两个类的所有功能

谢谢

0 个答案:

没有答案