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



下面是一个简化的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_

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

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

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

相关内容

  • 没有找到相关文章

最新更新