我使用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?
我的问题是如何获得两个类的所有功能
谢谢