如何在Python中打印列名称以及Logistic回归的列协同效应?

时间:2018-01-24 17:02:27

标签: python scikit-learn

logreg = LogisticRegression()

logreg.fit(X_train,y_train)

print(logreg.coef_ )
print(X.columns)

如何使用列名压缩logreg.coef_ long? 谢谢:)

2 个答案:

答案 0 :(得分:1)

这是一种方法:

# Create lists
column_labels = X.columns.tolist()
coef = logreg.coef_.squeeze().tolist()

# Zip together
labels_coef = list(zip(column_labels, coef))

# Verify the result
print(labels_coef)

答案 1 :(得分:0)

zip中的系数和特征

print(list(zip(X_train.columns.tolist(),logreg.coef_[0])))

DataFrame中的系数和特征

pd.DataFrame({"Feature":X_train.columns.tolist(),"Coefficients":logreg.coef_[0]})

enter image description here

相关问题