Sci-kit:使用GridSearchCV和Pipeline时,我可以访问学习者的属性,例如.coef_吗?

时间:2016-03-21 20:14:03

标签: python scikit-learn

这是LogisticRegression的简化示例。此版本(没有GridSearchCV和Pipeline)工作

clf = LogisticRegression()
clf.fit(X,y)
print clf.coef_

无法正常工作,因为GridSearchCV对象没有属性:

lr_pipeline = Pipeline([('clf', LogisticRegression())])
lr_parameters = {}
lr_gs = GridSearchCV(lr_pipeline, lr_parameters)
lr_gs = lr_gs.fit(X,y)
print lr_gs.coef_

有没有办法访问它? 谢谢!

1 个答案:

答案 0 :(得分:1)

首先需要对从网格搜索中获得的best_estimator_进行操作。然后,您需要访问管道中的clf步骤。然后你可以检索系数。

lr_gs.best_estimator_.named_steps['clf'].coef_