【发布时间】:2018-12-29 23:20:01
【问题描述】:
我有一个评论数据集,其类别标签为正面/负面。我正在对该评论数据集应用逻辑回归。首先,我正在转换成词袋。这里 sorted_data['Text'] 是 reviews 而 final_counts 是 稀疏矩阵
count_vect = CountVectorizer()
final_counts = count_vect.fit_transform(sorted_data['Text'].values)
standardized_data = StandardScaler(with_mean=False).fit_transform(final_counts)
将数据集拆分为训练和测试
X_1, X_test, y_1, y_test = cross_validation.train_test_split(final_counts, labels, test_size=0.3, random_state=0)
X_tr, X_cv, y_tr, y_cv = cross_validation.train_test_split(X_1, y_1, test_size=0.3)
我正在应用逻辑回归算法如下
optimal_lambda = 0.001000
log_reg_optimal = LogisticRegression(C=optimal_lambda)
# fitting the model
log_reg_optimal.fit(X_tr, y_tr)
# predict the response
pred = log_reg_optimal.predict(X_test)
# evaluate accuracy
acc = accuracy_score(y_test, pred) * 100
print('\nThe accuracy of the Logistic Regression for C = %f is %f%%' % (optimal_lambda, acc))
我的体重是
weights = log_reg_optimal.coef_ . #<class 'numpy.ndarray'>
array([[-0.23729528, -0.16050616, -0.1382504 , ..., 0.27291847,
0.35857267, 0.41756443]])
(1, 38178) #shape of weights
我想获得特征重要性,即;前 100 个具有高权重的特征。谁能告诉我如何获得它们?
【问题讨论】:
-
为什么不取
weights的绝对值,然后保持前100名?? -
是的。我还需要前 100 个具有高权重的单词。
标签: machine-learning scikit-learn logistic-regression sklearn-pandas